| /external/tensorflow/tensorflow/python/training/ |
| saver_large_variable_test.py | 52 }, write_version=saver_pb2.SaverDef.V1)
|
| checkpoint_management.py | 317 if format_version == saver_pb2.SaverDef.V2:
340 saver_pb2.SaverDef.V2)
342 saver_pb2.SaverDef.V1)
371 saver_pb2.SaverDef.V2)
413 saver_pb2.SaverDef.V2)
427 checkpoint_format_version=saver_pb2.SaverDef.V2,
435 checkpoint_format_version: `SaverDef.CheckpointFormatVersion`, defaults to
436 `SaverDef.V2`.
441 if checkpoint_format_version == saver_pb2.SaverDef.V2:
|
| saver.py | 84 def __init__(self, write_version=saver_pb2.SaverDef.V2):
113 if self._write_version == saver_pb2.SaverDef.V1:
119 elif self._write_version == saver_pb2.SaverDef.V2:
293 if self._write_version == saver_pb2.SaverDef.V2:
450 A SaverDef proto.
528 return saver_pb2.SaverDef(
553 return saver_pb2.SaverDef(
697 write_version=saver_pb2.SaverDef.V2,
752 saver_def: Optional `SaverDef` proto to use instead of running the
[all...] |
| checkpoint_management_test.py | 278 for version in (saver_pb2.SaverDef.V2, saver_pb2.SaverDef.V1):
297 for version in (saver_pb2.SaverDef.V2, saver_pb2.SaverDef.V1):
312 for version in (saver_pb2.SaverDef.V2, saver_pb2.SaverDef.V1):
|
| training.py | 142 tf_export(v1=["train.SaverDef"])(SaverDef)
|
| saver_test.py | 386 for ver in (saver_pb2.SaverDef.V1, saver_pb2.SaverDef.V2):
[all...] |
| /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/python/training/saving/ |
| functional_saver.py | 54 """Serializes to a SaverDef referencing the current graph."""
70 return saver_pb2.SaverDef(
74 version=saver_pb2.SaverDef.V2)
|
| /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
91 checkpoint_version=saver_pb2.SaverDef.V2):
96 input_saver_def: A `SaverDef` (optional).
117 checkpoint_version: Tensorflow variable file format (saver_pb2.SaverDef.V1
118 or saver_pb2.SaverDef.V2)
274 """Parses input tensorflow Saver into SaverDef proto."""
280 saver_def = saver_pb2.SaverDef()
303 checkpoint_version=saver_pb2.SaverDef.V2):
330 checkpoint_version: Tensorflow variable file format (saver_pb2.SaverDef.V1
331 or saver_pb2.SaverDef.V2)
[all...] |
| freeze_graph_test.py | 163 self._testFreezeGraph(saver_pb2.SaverDef.V1)
167 self._testFreezeGraph(saver_pb2.SaverDef.V2)
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| make_test_graphs.py | 54 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
69 saver = saver_lib.Saver(name='abcprefix', write_version=saver_pb2.SaverDef.V1)
76 # Without the SaverDef, the restore op won't be named correctly.
|
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item_builder.cc | 559 SaverDef saver;
576 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/slim/python/slim/ |
| evaluation_test.py | 250 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| variables.py | 748 write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| builder_impl.py | 222 write_version=saver_pb2.SaverDef.V2,
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_ops_test.py | [all...] |
| /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 | 301 saver_def: a SaverDef proto referencing Save/Restore ops to be retained.
576 saver_def: `SaverDef` protocol buffer.
606 if saver_def and not isinstance(saver_def, saver_pb2.SaverDef):
607 raise TypeError("saver_def must be of type SaverDef, not %s",
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| layout_optimizer_test.py | 238 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
[all...] |
| /external/tensorflow/tensorflow/python/layers/ |
| normalization_test.py | 64 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
[all...] |
