| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 210439649
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
This change provides an alternative mechanism to
tf.raise_exception_on_not_ok_status(), which is inefficient and
error-prone (people often use the status multiple times in the with
block, but it's only checked when the context manager exits). Instead,
it uses SWIG to automatically raise an exception when a C API method
fails. Note that this removes the status argument from affected
methods.
For now, I've only applied this typemap to C API methods. It would be
good to expand this to all uses of raise_exception_on_not_ok_status.
PiperOrigin-RevId: 191121016
|
|
|
|
| |
PiperOrigin-RevId: 181642475
|
|
|
|
|
|
|
| |
* add a bfloat16 Python type and NumPy extension.
* allow the bfloat16 type in a number places in the Python libraries.
PiperOrigin-RevId: 177875784
|
|
|
|
|
|
|
| |
This function is used extensively in imperative_grad. Implementing
it in C++ reduces SPINN training time by over 8%.
PiperOrigin-RevId: 172663591
|
|
|
|
| |
PiperOrigin-RevId: 170280771
|
|
|
|
| |
PiperOrigin-RevId: 164902588
|
|
|
|
| |
PiperOrigin-RevId: 160351254
|
|
|
|
|
|
| |
TF model.
PiperOrigin-RevId: 160162256
|
|
|
|
| |
PiperOrigin-RevId: 157178951
|
|
|
|
| |
Change: 154232702
|
|
|
|
|
|
| |
save op exists. This allows checkpointing of variables created by the rewriter.
Change: 151656461
|
|
|
|
| |
Change: 145009203
|
|
|
|
|
| |
instead now.
Change: 140548344
|
|
|
|
| |
Change: 138933733
|
|
|
|
|
|
| |
header for use with 'selective registration' (see
core/framework/selective_registration.h), given an input model.
Change: 134659033
|
|
|
|
|
| |
inference function. Enable this for MatMul and SparseMatMul.
Change: 131097313
|
|
|
|
|
| |
Also added client/net_lib to get access to PickUnusedPortOrDie.
Change: 129250139
|
|
|
|
|
|
| |
split out the SWIG API and the python code that calls it to enable easier testing.
Change: 127282792
|
|
|
|
|
|
|
|
| |
A simple example:
new_graph = pywrap_tensorflow.DoQuantizeTrainingOnGraphDef(8, sess.graph_def)
Then continue training with the new_graph.
Change: 124874712
|
|
|
|
| |
Change: 123900456
|
|
|
|
| |
Change: 122788931
|
|
|
|
|
|
|
|
|
| |
in Python.
- Add a tf_status util file to convert between Status and TF_Status.
- Use TF_Status for the swigged APIs in session, checkpoint_reader, and server_lib.
- Converts all these tf_status to exceptions in Python with the new context handler.
- Remove the old StatusNotOK exception.
Change: 121644982
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
This is a breaking change! The following classes have been renamed:
tf.GrpcServer -> tf.train.Server
tf.ClusterSpec -> tf.train.ClusterSpec
tf.ServerDef -> tf.train.ServerDef
tf.JobDef -> tf.train.JobDef
tf.ClusterDef -> tf.train.ClusterDef
The constructor for tf.train.Server is more permissive and now accepts
tf.train.ClusterSpec, tf.train.ClusterDef, and dictionary inputs for
specifying the cluster part of the server.
For consistency, the server library moves from python/client to python/training.
Change: 119335624
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
DebugString()
HasTensor()
GetVariableToShapeMap()
Example use can be found in PywrapTensorSliceReaderTest in
third_party/tensorflow/python/training/saver_test.py
For example, To use (in Python):
try:
reader = tf.train.NewCheckpointReader(your_checkpoint_file)
print(reader.DebugString())
except pywrap_tensorflow.StatusNotOK as e:
print(str(e))
Change: 117951901
|
|
|
|
| |
Change: 116767947
|
|
|
|
|
|
|
|
|
| |
Adds support for binding a TensorFlow server to any port, to support
single-process testing.
This interface is a work in progress. In particular, it supports
launching a server, but the support for clean shutdown is incomplete.
Change: 116593644
|
|
|
|
|
|
| |
imported.
Change: 110842260
|
|
|
|
|
|
|
|
|
| |
Changes:
- Updates to docs
- Several changes for Python 3 compatibility
- Added license headers
Base CL: 108710566
|
|
TensorFlow is an open source software library for numerical computation
using data flow graphs.
Base CL: 107276108
|