diff options
author | 2017-12-01 16:38:52 -0800 | |
---|---|---|
committer | 2017-12-01 16:46:20 -0800 | |
commit | e30b0babce133631b19de1fd7bacc84c884d6f55 (patch) | |
tree | 2b034ed6283511f192760ed8c84e7870c7d29f9f /tensorflow/python/util/py_checkpoint_reader.i | |
parent | 4173ac695eb7b41fc1885b08ed52904dbf2278c3 (diff) |
Adding tf_export decorators to public API functions defined in swig files.
PiperOrigin-RevId: 177657932
Diffstat (limited to 'tensorflow/python/util/py_checkpoint_reader.i')
-rw-r--r-- | tensorflow/python/util/py_checkpoint_reader.i | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/util/py_checkpoint_reader.i b/tensorflow/python/util/py_checkpoint_reader.i index 0cd095d9d9..8004898cbc 100644 --- a/tensorflow/python/util/py_checkpoint_reader.i +++ b/tensorflow/python/util/py_checkpoint_reader.i @@ -164,6 +164,8 @@ def NewCheckpointReader(filepattern): with errors.raise_exception_on_not_ok_status() as status: from tensorflow.python.util import compat return CheckpointReader(compat.as_bytes(filepattern), status) + +NewCheckpointReader._tf_api_names = ['train.NewCheckpointReader'] %} %include "tensorflow/c/checkpoint_reader.h" |