diff options
author | Sherry Moore <sherrym@google.com> | 2016-04-01 12:18:42 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-04-01 13:22:07 -0700 |
commit | 286c5a48212d9eee522e026302524537e0e1cda8 (patch) | |
tree | 60ee9d215c0edb74fdf219f3f970f32132c296dd /tensorflow/python/util/py_checkpoint_reader.i | |
parent | cd762c0b0b77e8062237407ce678c8d576d77024 (diff) |
Use all PEP8 names for CheckpointReader.
The new APIs are
debug_string() : returns debug string
has_tensor(tensor_name): returns bool
get_tensor(tensor_name) : returns numpy array for tensor
get_variable_to_shape_map() : returns name to shape map
Change: 118806412
Diffstat (limited to 'tensorflow/python/util/py_checkpoint_reader.i')
-rw-r--r-- | tensorflow/python/util/py_checkpoint_reader.i | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/python/util/py_checkpoint_reader.i b/tensorflow/python/util/py_checkpoint_reader.i index 471a6b6be0..96ed0cfe61 100644 --- a/tensorflow/python/util/py_checkpoint_reader.i +++ b/tensorflow/python/util/py_checkpoint_reader.i @@ -107,8 +107,8 @@ limitations under the License. %unignore tensorflow::checkpoint; %unignore tensorflow::checkpoint::CheckpointReader; %unignore tensorflow::checkpoint::CheckpointReader::~CheckpointReader; -%unignore tensorflow::checkpoint::CheckpointReader::DebugString; -%unignore tensorflow::checkpoint::CheckpointReader::GetVariableToShapeMap; +%rename("debug_string") tensorflow::checkpoint::CheckpointReader::DebugString; +%rename("get_variable_to_shape_map") tensorflow::checkpoint::CheckpointReader::GetVariableToShapeMap; %rename("_HasTensor") tensorflow::checkpoint::CheckpointReader::HasTensor; %unignore tensorflow::checkpoint::PyCheckpointReader; %unignore tensorflow::checkpoint::~PyCheckpointReader; @@ -134,7 +134,6 @@ limitations under the License. return _NewPyCheckpointReader(compat.as_bytes(filepattern)) %} - %newobject tensorflow::checkpoint::PyCheckpointReader::get_tensor; %extend tensorflow::checkpoint::PyCheckpointReader { |