Home
last modified time | relevance | path

Searched refs:saver (Results 1 – 25 of 245) sorted by relevance

12345678910

/external/tensorflow/tensorflow/python/training/
Dsession_manager_test.py31 from tensorflow.python.training import saver as saver_lib
99 saver = saver_lib.Saver({"v": v})
103 saver=saver,
108 saver.save(sess, checkpoint_filename)
119 saver = saver_lib.Saver({"v": v})
126 saver=saver,
137 saver=saver,
156 saver = saver_lib.Saver({"v": v})
159 saver=saver,
182 saver = saver_lib.Saver({"v": v})
[all …]
Dsession_manager.py46 sess, saver, path): argument
66 saver.restore(sess, path)
181 saver=None, argument
224 if not saver or not (checkpoint_dir or checkpoint_filename_with_path):
229 sess, saver, checkpoint_filename_with_path)
246 sess, saver, ckpt.model_checkpoint_path)
247 saver.recover_last_checkpoints(ckpt.all_model_checkpoint_paths)
253 saver=None, argument
311 saver,
344 saver=None, argument
[all …]
Dsupervisor.py32 from tensorflow.python.training import saver as saver_mod
215 saver=USE_DEFAULT, argument
314 self._init_saver(saver=saver)
453 def _init_saver(self, saver=USE_DEFAULT): argument
460 if saver is Supervisor.USE_DEFAULT:
461 saver = self._get_first_op_from_collection(ops.GraphKeys.SAVERS)
462 if saver is None and variables.global_variables():
463 saver = saver_mod.Saver()
464 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
465 self._saver = saver
[all …]
Dquantize_training_test.py28 from tensorflow.python.training import saver as saver_module
65 saver = saver_module.Saver({'b': b})
80 saver.save(sess, save_path)
87 saver.restore(sess, save_path)
Devaluation_test.py37 from tensorflow.python.training import saver
118 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
150 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
173 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
201 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
230 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
/external/tensorflow/tensorflow/python/saved_model/
Dbuilder_impl.py35 from tensorflow.python.training import saver as tf_saver
219 def _maybe_create_saver(self, saver=None): argument
221 if not saver:
224 saver = tf_saver.Saver(
229 return saver
238 saver=None): argument
285 saver = self._maybe_create_saver(saver)
294 meta_graph_def = saver.export_meta_graph(
312 saver=None): argument
366 saver = self._maybe_create_saver(saver)
[all …]
Dload_v1_in_v2.py38 from tensorflow.python.training import saver as tf_saver
94 saver, _ = tf_saver._import_meta_graph_with_return_elements(
97 returns[0] = saver
99 def _extract_saver_restore(self, wrapped, saver): argument
100 if saver is None:
102 saver_def = saver.saver_def
230 saver, = load_graph_returns
231 restore_from_saver = self._extract_saver_restore(wrapped, saver)
Dloader_impl.py35 from tensorflow.python.training import saver as tf_saver
425 def restore_variables(self, sess, saver, import_scope=None): argument
443 if (saver is None and
447 elif isinstance(saver, tf_saver.Saver):
448 saver.restore(sess, self._variables_path)
495 saver, _ = self.load_graph(sess.graph, tags, import_scope,
497 self.restore_variables(sess, saver, import_scope)
Dloader_test.py34 from tensorflow.python.training import saver as tf_saver
146 saver, _ = loader.load_graph(
153 loader.restore_variables(sess, saver)
201 saver, _ = loader.load_graph(graph, ["foo_graph"])
203 loader.restore_variables(sess, saver)
262 saver=tf_saver.Saver([x, y]))
267 saver, _ = loader.load_graph(sess.graph, ["foo_graph"])
269 self.assertIsNotNone(saver)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmake_saveable_from_iterator_test.py26 from tensorflow.python.training import saver as saver_lib
48 saver = saver_lib.Saver()
49 return init_ops, get_next_ops, saver
61 init_ops, get_next_ops, saver = self._build_graph(num_pipelines,
69 saver.save(sess, self._ckpt_path())
72 init_ops, get_next_ops, saver = self._build_graph(num_pipelines,
76 saver.restore(sess, self._ckpt_path())
90 _, _, saver = self._build_graph(num_pipelines, num_outputs)
93 saver.save(sess, self._ckpt_path())
/external/tensorflow/tensorflow/python/checkpoint/
Dfunctional_saver_test.py56 saver = functional_saver.MultiDeviceSaver(
59 self.evaluate(saver.save(constant_op.constant(prefix)))
62 self.evaluate(saver.restore(prefix))
76 saver = functional_saver.MultiDeviceSaver(
79 self.evaluate(saver.save(constant_op.constant(prefix), self.local_options))
82 self.evaluate(saver.restore(prefix, self.local_options))
101 saver = functional_saver.MultiDeviceSaver(
107 lambda: proto_accumulator.append(saver.to_proto()), signature=())
141 saver = functional_saver.MultiDeviceSaver(
146 self.evaluate(saver.save(constant_op.constant(prefix)))
[all …]
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test_base.py36 from tensorflow.python.training import saver as saver_lib
276 saver = self._import_meta_graph()
282 self._restore(saver, sess)
319 init_op, get_next_op, saver = self._build_graph(
327 self._save(sess, saver)
432 saver = self._import_meta_graph()
436 init_op, get_next_op, saver = self._build_graph(
438 return init_op, get_next_op, saver
442 init_op, get_next_op, saver = get_ops()
447 self._restore(saver, sess)
[all …]
/external/tensorflow/tensorflow/python/trackable/
Dpython_state_test.py148 saver = util.Checkpoint(numpy=save_state)
156 first_save_path = saver.save(prefix)
159 second_save_path = saver.save(prefix)
182 saver = util.Checkpoint(numpy=save_state)
184 save_path = saver.save(prefix)
185 saver.restore(save_path)
187 saver.save(prefix)
189 saver.save(prefix)
190 saver.restore(save_path)
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_values_test.py35 from tensorflow.python.training import saver as saver_lib
114 saver = saver_lib.Saver(var_list=[var])
117 return saver.save(sess, prefix), saver
164 saver = saver_lib.Saver(var_list=[var])
165 saver.restore(sess, save_path)
178 saver = saver_lib.Saver(var_list=[mirrored])
179 saver.restore(sess, save_path)
192 save_path, saver = self._save_return_saver(sess, mirrored)
198 saver.restore(sess, save_path)
Dvalues_test.py45 from tensorflow.python.training import saver as saver_lib
449 saver = saver_lib.Saver(var_list=[var])
452 return saver.save(sess, prefix), saver
615 save_path, saver = self._save_return_saver(sess, replica_local)
622 saver.restore(sess, save_path)
639 save_path, saver = self._save_return_saver(sess, replica_local)
645 saver.restore(sess, save_path)
708 saver = saver_lib.Saver(var_list=[var])
709 saver.restore(sess, save_path)
723 saver = saver_lib.Saver(var_list=[replica_local])
[all …]
Dwarm_starting_util_test.py34 from tensorflow.python.training import saver as saver_lib
65 saver = saver_lib.Saver()
67 saver.save(sess, ckpt_prefix, global_step=0)
/external/spdx-tools/docs/
Djsonsaver.md5 …is converted to JSON using a single json.MarshalIndent function call. The saver uses temporary mem…
11 …ich is an array of the SPDX Identifiers of the files of that package. The saver iterates through t…
22 * The saver does not make any pre-assumptions based on the order in which the properties are saved.
25 …es not define the SPDX Identifier property for the annotation object. The saver inserts the annota…
/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint.py33 from tensorflow.python.training import saver as saver_lib
46 saver: Optional[saver_lib.Saver] = None,
73 if saver is not None and scaffold is not None:
75 self._saver = saver
198 return self._scaffold.saver
Dsession_support.py320 def __init__(self, checkpoint_prefix, saver=None, on_shutdown_hooks=None): argument
321 self._saver = saver
334 if training_util.get_global_step() is None and self.saver() is not None:
359 def saver(self): member in GracefulShutdownHook
389 if self.saver():
392 self.saver().save(
/external/tensorflow/tensorflow/python/tools/
Dfreeze_graph_test.py44 from tensorflow.python.training import saver as saver_lib
66 saver = saver_lib.Saver(write_version=saver_write_version)
67 checkpoint_path = saver.save(
182 saver = saver_lib.Saver()
183 checkpoint_path = saver.save(
305 saver = saver_lib.Saver()
306 checkpoint_path = saver.save(
/external/tensorflow/tensorflow/compiler/aot/tests/
Dmake_test_graphs.py38 from tensorflow.python.training import saver as saver_lib
55 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
61 saver.save(sess, ckpt)
70 saver = saver_lib.Saver(name='abcprefix', write_version=saver_pb2.SaverDef.V1)
76 saver.save(sess, ckpt_file)
80 f.write(six.ensure_binary(saver.as_saver_def().SerializeToString()))
/external/tensorflow/tensorflow/python/ops/
Dboosted_trees_ops.py48 from tensorflow.python.training import saver
67 class QuantileAccumulatorSaveable(saver.BaseSaverBuilder.SaveableObject):
80 return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix)
162 class _TreeEnsembleSavable(saver.BaseSaverBuilder.SaveableObject):
180 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
182 saver.BaseSaverBuilder.SaveSpec(serialized, slice_spec,
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc593 SaverDef saver; in GrapplerItemFromMetaGraphDef() local
596 if (!saver.ParseFromString(raw)) { in GrapplerItemFromMetaGraphDef()
599 if (saver.filename_tensor_name().empty()) { in GrapplerItemFromMetaGraphDef()
602 new_item->save_op = saver.save_tensor_name(); in GrapplerItemFromMetaGraphDef()
603 new_item->restore_op = saver.restore_op_name(); in GrapplerItemFromMetaGraphDef()
604 new_item->save_restore_loc_tensor = saver.filename_tensor_name(); in GrapplerItemFromMetaGraphDef()
610 const SaverDef& saver = meta_graph.saver_def(); in GrapplerItemFromMetaGraphDef() local
611 new_item->save_op = saver.save_tensor_name(); in GrapplerItemFromMetaGraphDef()
612 new_item->restore_op = saver.restore_op_name(); in GrapplerItemFromMetaGraphDef()
613 new_item->save_restore_loc_tensor = saver.filename_tensor_name(); in GrapplerItemFromMetaGraphDef()
/external/angle/third_party/abseil-cpp/absl/random/internal/
Diostream_state_saver_test.cc36 auto saver = make_ostream_state_saver(ss); in StreamRoundTrip() local
42 auto saver = make_istream_state_saver(ss); in StreamRoundTrip() local
60 auto saver = make_ostream_state_saver(ss); in StreamRoundTrip() local
66 auto saver = make_istream_state_saver(ss); in StreamRoundTrip() local
86 auto saver = make_ostream_state_saver(ss); in TEST() local
/external/webrtc/third_party/abseil-cpp/absl/random/internal/
Diostream_state_saver_test.cc36 auto saver = make_ostream_state_saver(ss); in StreamRoundTrip() local
42 auto saver = make_istream_state_saver(ss); in StreamRoundTrip() local
60 auto saver = make_ostream_state_saver(ss); in StreamRoundTrip() local
66 auto saver = make_istream_state_saver(ss); in StreamRoundTrip() local
86 auto saver = make_ostream_state_saver(ss); in TEST() local

12345678910