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

1 files changed, 8 insertions, 0 deletions

diff --git a/tensorflow/core/util/tensor_slice_reader.h b/tensorflow/core/util/tensor_slice_reader.h
index 19f8bd068a..5dd0b2f919 100644
--- a/tensorflow/core/util/tensor_slice_reader.h
+++ b/tensorflow/core/util/tensor_slice_reader.h
@@ -62,6 +62,7 @@ class TensorSliceReader {
   typedef std::function<Status(const string&, Table**)> OpenTableFunction;
 
   static const int kLoadAllShards = -1;
+  TensorSliceReader(const string& filepattern);
   TensorSliceReader(const string& filepattern, OpenTableFunction open_function);
   TensorSliceReader(const string& filepattern, OpenTableFunction open_function,
                     int preferred_shard);
@@ -95,6 +96,13 @@ class TensorSliceReader {
     return tensors_;
   }
 
+  typedef std::unordered_map<string, TensorShape> VarToShapeMap;
+  // Returns a map from tensor name to shape.
+  VarToShapeMap GetVariableToShapeMap() const;
+
+  // Returns a string containing names and shapes of all the tensors.
+  const string DebugString() const;
+
  private:
   friend class TensorSliceWriteTestHelper;