| /external/tensorflow/tensorflow/python/training/ |
| saver_large_variable_test.py | 52 }, write_version=saver_pb2.SaverDef.V1)
|
| saver.py | 202 def __init__(self, write_version=saver_pb2.SaverDef.V2):
231 if self._write_version == saver_pb2.SaverDef.V1:
237 elif self._write_version == saver_pb2.SaverDef.V2:
410 if self._write_version == saver_pb2.SaverDef.V2:
741 A SaverDef proto.
810 return saver_pb2.SaverDef(
[all...] |
| saver_test.py | 341 for ver in (saver_pb2.SaverDef.V1, saver_pb2.SaverDef.V2):
749 _WRITE_VERSION = saver_pb2.SaverDef.V1
782 if save._write_version is saver_pb2.SaverDef.V1:
789 if save._write_version is saver_pb2.SaverDef.V1:
[all...] |
| training.py | 239 "SaverDef", # From saver_pb2.
255 tf_export("train.SaverDef")(SaverDef)
|
| /external/tensorflow/tensorflow/python/tools/ |
| freeze_graph.py | 17 This script is designed to take a GraphDef proto, a SaverDef proto, and a set of
76 checkpoint_version=saver_pb2.SaverDef.V2):
197 """Parser input tensorflow Saver into SaverDef proto."""
203 saver_def = saver_pb2.SaverDef()
226 checkpoint_version=saver_pb2.SaverDef.V2):
261 checkpoint_version = saver_pb2.SaverDef.V1
263 checkpoint_version = saver_pb2.SaverDef.V2
|
| freeze_graph_test.py | 161 self._testFreezeGraph(saver_pb2.SaverDef.V1)
164 self._testFreezeGraph(saver_pb2.SaverDef.V2)
|
| /external/tensorflow/tensorflow/contrib/rnn/python/tools/ |
| checkpoint_convert.py | 242 write_version = (saver_pb2.SaverDef.V1
243 if write_v1_checkpoint else saver_pb2.SaverDef.V2)
|
| /external/tensorflow/tensorflow/contrib/session_bundle/example/ |
| export_half_plus_two.py | 69 write_version=tf.train.SaverDef.V2 if use_checkpoint_v2 else
70 tf.train.SaverDef.V1)
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| make_test_graphs.py | 52 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
67 saver = saver_lib.Saver(name='abcprefix', write_version=saver_pb2.SaverDef.V1)
74 # Without the SaverDef, the restore op won't be named correctly.
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| builder_impl.py | 299 write_version=saver_pb2.SaverDef.V2,
387 write_version=saver_pb2.SaverDef.V2,
|
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item_builder.cc | 429 SaverDef saver;
446 const SaverDef& saver = meta_graph.saver_def();
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| exporter_test.py | 109 write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_ops_test.py | 217 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
269 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
313 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| evaluation_test.py | 239 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| variables.py | 689 write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| meta_graph_transform_test.py | 139 saver_def = saver_pb2.SaverDef()
|
| /external/tensorflow/tensorflow/python/framework/ |
| meta_graph.py | 299 saver_def: a SaverDef proto referencing Save/Restore ops to be retained.
526 saver_def: `SaverDef` protocol buffer.
556 if saver_def and not isinstance(saver_def, saver_pb2.SaverDef):
557 raise TypeError("saver_def must be of type SaverDef, not %s",
843 saver_def: `SaverDef` protocol buffer.
846 with the provided SaverDef
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| layout_optimizer_test.py | 234 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
[all...] |
| /external/tensorflow/tensorflow/python/layers/ |
| normalization_test.py | 62 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
[all...] |
