diff options
| author |  Zongheng Yang <zongheng@google.com> | 2016-10-14 21:10:19 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-10-14 22:18:59 -0700 |
| commit | 1e324a0f2a67cfa651677bd381bf1bf2adc3e2f8 (patch) | |
| tree | 28e83ed5affb60d00426540458d930e79b5e67dc /tensorflow/c/checkpoint_reader.h | |
| parent | c53f1a9930b85110bb3164695f853722ce293d29 (diff) | |
checkpoint_reader: fix VarToShapeMap V2 impl.
This change makes it adhere to the original semantics: all slices of a
partitioned tensor are grouped under one entry.
Change: 136229541
Diffstat (limited to 'tensorflow/c/checkpoint_reader.h')
| -rw-r--r-- | tensorflow/c/checkpoint_reader.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/c/checkpoint_reader.h b/tensorflow/c/checkpoint_reader.h index 6d27bdc375..1b09dcd18d 100644 --- a/tensorflow/c/checkpoint_reader.h +++ b/tensorflow/c/checkpoint_reader.h @@ -43,6 +43,8 @@ class CheckpointReader { bool HasTensor(const string& name) const; const string DebugString() const; + // Returns a map from variable namaes to its shape. Slices of a partitioned + // tensor are combined into a single entry. const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const; // Attempts to look up the tensor named "name" and stores the found result in |