| /external/tensorflow/tensorflow/python/training/ |
| saver.py | 693 saver_def=None,
752 saver_def: Optional `SaverDef` proto to use instead of running the
755 The `saver_def` proto should be the one returned by the
757 builder: Optional `SaverBuilder` to use if a `saver_def` was not provided.
[all...] |
| basic_session_run_hooks.py | 553 # We do write graph and saver_def at the first call of before_run.
560 saver_def = self._get_saver().saver_def if self._get_saver() else None
564 saver_def=saver_def)
[all...] |
| supervisor_test.py | 775 saver_def=sv.saver.saver_def)
825 saver_def=sv.saver.saver_def)
[all...] |
| supervisor.py | 326 saver_def=self._saver.saver_def if self._saver else None)
[all...] |
| basic_session_run_hooks_test.py | 750 saver_def=self.scaffold.saver.saver_def)
[all...] |
| saver_test.py | 381 save.saver_def.filename_tensor_name)
[all...] |
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| meta_graph_transform.py | 103 def _freeze_transform(graph_def, output_names, initializer_names, saver_def,
116 saver_def: A SaverDef proto used for restoring a checkpoint during freezing,
130 saver_def, checkpoint_path)
272 saver_def=None,
288 saver_def: A SaverDef proto used for restoring a checkpoint during freezing,
311 transformed_initializer_names, saver_def, checkpoint_path)
378 saver = _saver_lib.Saver(saver_def=input_saver_def)
410 if base_meta_graph_def.HasField('saver_def'):
411 saver_defs.append(base_meta_graph_def.saver_def)
414 for saver_def in saver_defs
[all...] |
| meta_graph_transform_test.py | 139 saver_def = saver_pb2.SaverDef()
140 saver_def.filename_tensor_name = 'f1'
144 initializer_names, saver_def,
154 table_init_names, main_op, saver_def,
411 base_meta_graph_def.saver_def.filename_tensor_name = 'node1'
412 base_meta_graph_def.saver_def.save_tensor_name = 'node3'
413 base_meta_graph_def.saver_def.restore_op_name = 'node6'
422 group_deps_name, base_meta_graph_def.saver_def, None)
539 base_meta_graph_def.saver_def.filename_tensor_name = 'node1'
540 base_meta_graph_def.saver_def.save_tensor_name = 'node3
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| meta_graph.py | 292 def _find_extraneous_saver_nodes(graph_def, saver_def):
301 saver_def: a SaverDef proto referencing Save/Restore ops to be retained.
318 if saver_def is not None:
319 save_op_name = _op_name(saver_def.save_tensor_name)
320 restore_op_name = _op_name(saver_def.restore_op_name)
563 saver_def=None,
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...] |
| meta_graph_test.py | 93 saver_def=None)
[all...] |
| /external/tensorflow/tensorflow/python/saved_model/ |
| load_v1_in_v2.py | 90 saver_def = saver.saver_def
93 saver_def.filename_tensor_name)],
94 fetches=[wrapped.graph.as_graph_element(saver_def.restore_op_name)])
|
| save.py | 532 saver_def = saver.to_proto()
533 meta_graph_def.saver_def.CopyFrom(saver_def)
[all...] |
| saved_model_test.py | [all...] |
| /external/tensorflow/tensorflow/python/tpu/ |
| async_checkpoint.py | 100 # We do write graph and saver_def at the first call of before_run.
111 saver_def = self._get_saver().saver_def if self._get_saver() else None
114 graph_def=graph.as_graph_def(add_shapes=True), saver_def=saver_def)
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| exporter.py | 146 saver_def = saver.as_saver_def()
147 saver_def.ClearField("max_to_keep")
148 self._saver = tf_saver.Saver(saver_def=saver_def)
|
| session_bundle.cc | 209 bundle->meta_graph_def.saver_def().restore_op_name(),
210 bundle->meta_graph_def.saver_def().filename_tensor_name(),
|
| /external/tensorflow/tensorflow/python/tools/ |
| freeze_graph.py | 150 saver_def=input_saver_def, write_version=checkpoint_version)
280 saver_def = saver_pb2.SaverDef()
282 saver_def.ParseFromString(f.read())
284 text_format.Merge(f.read(), saver_def)
285 return saver_def
|
| /external/tensorflow/tensorflow/cc/saved_model/ |
| loader.cc | 276 bundle->meta_graph_def.saver_def().restore_op_name(),
277 bundle->meta_graph_def.saver_def().filename_tensor_name(),
|
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item_builder.cc | 576 const SaverDef& saver = meta_graph.saver_def();
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| estimator_test.py | 496 wraps=real_saver, saver_def=real_saver.saver_def)
[all...] |
