diff options
| author |  Sherry Moore <sherrym@google.com> | 2016-03-23 10:40:08 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-03-23 12:12:12 -0700 |
| commit | 2966fcdae0cf7d75f4eca027f3003e355089b0ed (patch) | |
| tree | 0b815993c33dd25b5d8c32ace77eb9e0431367f5 /tensorflow/python/tensorflow.i | |
| parent | f10637b372b3216400129b42abacd4fe50a6d7ea (diff) | |
Added Python APIs for inspecting checkpoint content.
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
Diffstat (limited to 'tensorflow/python/tensorflow.i')
| -rw-r--r-- | tensorflow/python/tensorflow.i | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/tensorflow.i b/tensorflow/python/tensorflow.i index fb2dd00b69..5ff8cf3fe3 100644 --- a/tensorflow/python/tensorflow.i +++ b/tensorflow/python/tensorflow.i @@ -18,6 +18,7 @@ limitations under the License. * includes follows dependency order */ %include "tensorflow/python/util/port.i" +%include "tensorflow/python/util/py_checkpoint_reader.i" %include "tensorflow/python/lib/core/py_func.i" %include "tensorflow/python/lib/core/status.i" |