Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/contrib/eager/python/
Dsaver_test.py22 from tensorflow.contrib.eager.python import saver as _saver
53 saver = _saver.Saver([v1])
54 saver.save(ckpt_prefix)
58 saver.restore(ckpt_prefix)
71 saver = _saver.Saver([v1_first_graph, v1_second_graph])
74 saver.save(ckpt_prefix)
81 saver = _saver.Saver([v1, v1])
83 saver.save(ckpt_prefix)
96 saver = _saver.Saver({'ckpt/v1': v1, 'ckpt/v2': v2})
97 saver.save(ckpt_prefix)
[all …]
/external/tensorflow/tensorflow/python/training/
Dsession_manager_test.py33 from tensorflow.python.training import saver as saver_lib
84 saver = saver_lib.Saver({"v": v})
88 saver=saver,
93 saver.save(sess, checkpoint_filename)
104 saver = saver_lib.Saver({"v": v})
111 saver=saver,
122 saver=saver,
141 saver = saver_lib.Saver({"v": v})
144 saver=saver,
167 saver = saver_lib.Saver({"v": v})
[all …]
Dsession_manager.py27 from tensorflow.python.training import saver as saver_mod
152 saver=None, argument
187 if not saver or not (checkpoint_dir or checkpoint_filename_with_path):
191 saver.restore(sess, checkpoint_filename_with_path)
207 saver.restore(sess, ckpt.model_checkpoint_path)
208 saver.recover_last_checkpoints(ckpt.all_model_checkpoint_paths)
214 saver=None, argument
270 saver,
303 saver=None, argument
335 saver,
Dtraining.py168 from tensorflow.python.training.saver import Saver
169 from tensorflow.python.training.saver import checkpoint_exists
170 from tensorflow.python.training.saver import generate_checkpoint_state_proto
171 from tensorflow.python.training.saver import get_checkpoint_mtimes
172 from tensorflow.python.training.saver import get_checkpoint_state
173 from tensorflow.python.training.saver import latest_checkpoint
174 from tensorflow.python.training.saver import update_checkpoint_state
175 from tensorflow.python.training.saver import export_meta_graph
176 from tensorflow.python.training.saver import import_meta_graph
Dsupervisor.py36 from tensorflow.python.training import saver as saver_mod
219 saver=USE_DEFAULT, argument
318 self._init_saver(saver=saver)
456 def _init_saver(self, saver=USE_DEFAULT): argument
463 if saver is Supervisor.USE_DEFAULT:
464 saver = self._get_first_op_from_collection(ops.GraphKeys.SAVERS)
465 if saver is None and variables.global_variables():
466 saver = saver_mod.Saver()
467 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
468 self._saver = saver
[all …]
Dquantize_training_test.py31 from tensorflow.python.training import saver as saver_module
66 saver = saver_module.Saver({'b': b})
81 saver.save(sess, save_path)
88 saver.restore(sess, save_path)
Devaluation_test.py41 from tensorflow.python.training import saver
123 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
154 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/contrib/learn/python/learn/utils/
Dexport.py34 from tensorflow.python.training import saver as tf_saver
50 saver = _get_first_op_from_collection(ops.GraphKeys.SAVERS)
51 if saver is not None:
52 if saver:
53 saver = saver[0]
55 saver = None
56 if saver is None and variables.global_variables():
57 saver = tf_saver.Saver()
58 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
59 return saver
[all …]
Dexport_test.py39 from tensorflow.python.training import saver
57 save = saver.import_meta_graph(export_meta_filename)
72 saver.checkpoint_exists(os.path.join(export_dir, '00000001', 'export')),
76 saver.checkpoint_exists(os.path.join(export_dir, '00000010', 'export')),
245 saver.checkpoint_exists(os.path.join(export_dir, '00000000', 'export'))
247 saver.checkpoint_exists(os.path.join(export_dir, '00000010', 'export')))
/external/tensorflow/tensorflow/contrib/slim/python/slim/
Devaluation.py128 from tensorflow.python.training import saver as tf_saver
198 saver = None
200 saver = tf_saver.Saver(variables_to_restore)
206 init_op=initial_op, init_feed_dict=initial_op_feed_dict, saver=saver),
283 saver = None
285 saver = tf_saver.Saver(variables_to_restore)
292 init_fn=init_fn, saver=saver),
Dlearning.py269 from tensorflow.python.training import saver as tf_saver
550 saver=None, argument
638 if saver is not None:
658 saver = saver or tf_saver.Saver()
733 saver=saver,
780 sv.saver.save(sess, sv.save_path, global_step=sv.global_step)
Devaluation_test.py46 from tensorflow.python.training import saver as saver_lib
103 saver = saver_lib.Saver()
106 saver.save(sess, os.path.join(chkpt_dir, 'chkpt'))
202 saver = saver_lib.Saver()
205 saver.save(sess, os.path.join(chkpt_dir, 'chkpt'))
239 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
242 saver.save(sess, checkpoint_path)
/external/tensorflow/tensorflow/contrib/data/python/kernel_tests/
Ddataset_serialization_test_base.py35 from tensorflow.python.training import saver as saver_lib
250 saver = self._import_meta_graph()
255 self._restore(saver, sess)
314 _, get_next_op, saver = self._build_graph(
318 self._restore(saver, sess)
375 get_next_op, saver = self._build_empty_graph(
379 self._restore(saver, sess)
409 init_op, get_next_op, saver = self._build_graph(
417 self._save(sess, saver)
501 saver = self._import_meta_graph()
[all …]
Dserialization_integration_test.py26 from tensorflow.python.training import saver as saver_lib
48 saver = saver_lib.Saver()
49 return init_ops, get_next_ops, saver
60 init_ops, get_next_ops, saver = self._build_graph(num_pipelines,
68 saver.save(sess, self._ckpt_path())
71 init_ops, get_next_ops, saver = self._build_graph(num_pipelines,
74 saver.restore(sess, self._ckpt_path())
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dgraph_actions.py47 from tensorflow.python.training import saver as tf_saver
95 def _restore_from_checkpoint(session, graph, checkpoint_path, saver=None): argument
97 saver = saver or _make_saver(graph)
98 if saver:
99 saver.restore(session, checkpoint_path)
302 saver=_make_saver(graph, keep_checkpoint_max),
383 supervisor.saver.save(session, ckpt_path, global_step=last_step)
416 saver = _get_first_op_from_collection(ops.GraphKeys.SAVERS)
417 if saver is None and variables.global_variables():
418 saver = tf_saver.Saver()
[all …]
/external/tensorflow/tensorflow/contrib/rnn/python/tools/
Dcheckpoint_convert_test.py29 from tensorflow.python.training import saver as saver_lib
62 saver = saver_lib.Saver()
64 saver.save(sess, self._old_ckpt_path)
77 saver = saver_lib.Saver()
79 saver.save(sess, self._old_ckpt_path)
92 saver = saver_lib.Saver()
94 saver.save(sess, self._old_ckpt_path)
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dstats_accumulator_ops.py28 from tensorflow.python.training import saver
34 class StatsAccumulator(saver.BaseSaverBuilder.SaveableObject):
95 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
97 saver.BaseSaverBuilder.SaveSpec(num_updates, slice_spec,
99 saver.BaseSaverBuilder.SaveSpec(partition_ids, slice_spec,
101 saver.BaseSaverBuilder.SaveSpec(feature_ids, slice_spec,
103 saver.BaseSaverBuilder.SaveSpec(gradients, slice_spec,
105 saver.BaseSaverBuilder.SaveSpec(hessians, slice_spec,
Dmodel_ops.py32 from tensorflow.python.training import saver
39 class TreeEnsembleVariableSavable(saver.BaseSaverBuilder.SaveableObject):
56 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
58 saver.BaseSaverBuilder.SaveSpec(ensemble_config, slice_spec,
/external/tensorflow/tensorflow/contrib/training/python/training/
Dtraining_test.py41 from tensorflow.python.training import saver as saver_lib
333 saver = saver_lib.Saver()
342 logdir, save_steps=50, saver=saver),
386 saver = saver_lib.Saver()
392 logdir1, save_steps=1, saver=saver),
403 saver = saver_lib.Saver()
409 logdir1, save_steps=300, saver=saver),
466 saver = saver_lib.Saver()
472 logdir, save_steps=200, saver=saver),
490 saver = saver_lib.Saver()
[all …]
Devaluation.py150 from tensorflow.python.training import saver as tf_saver
317 def _scaffold_with_init(scaffold, saver, checkpoint_path): argument
331 saver.restore(session, checkpoint_path)
341 saver=scaffold.saver)
/external/tensorflow/tensorflow/contrib/data/python/ops/
Diterator_ops.py22 from tensorflow.python.training import saver
63 class _Saveable(saver.BaseSaverBuilder.SaveableObject):
69 saver.BaseSaverBuilder.SaveSpec(serialized_iterator, "",
/external/tensorflow/tensorflow/compiler/aot/tests/
Dmake_test_graphs.py35 from tensorflow.python.training import saver as saver_lib
52 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
58 saver.save(sess, ckpt)
67 saver = saver_lib.Saver(name='abcprefix', write_version=saver_pb2.SaverDef.V1)
73 saver.save(sess, ckpt_file)
77 f.write(saver.as_saver_def().SerializeToString())
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dlearning.py78 saver=None, argument
182 saver=saver,
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc429 SaverDef saver; in GrapplerItemFromMetaGraphDef() local
432 if (!saver.ParseFromString(raw)) { in GrapplerItemFromMetaGraphDef()
435 if (saver.filename_tensor_name().empty()) { in GrapplerItemFromMetaGraphDef()
438 new_item->save_op = saver.save_tensor_name(); in GrapplerItemFromMetaGraphDef()
439 new_item->restore_op = saver.restore_op_name(); in GrapplerItemFromMetaGraphDef()
440 new_item->save_restore_loc_tensor = saver.filename_tensor_name(); in GrapplerItemFromMetaGraphDef()
446 const SaverDef& saver = meta_graph.saver_def(); in GrapplerItemFromMetaGraphDef() local
447 new_item->save_op = saver.save_tensor_name(); in GrapplerItemFromMetaGraphDef()
448 new_item->restore_op = saver.restore_op_name(); in GrapplerItemFromMetaGraphDef()
449 new_item->save_restore_loc_tensor = saver.filename_tensor_name(); in GrapplerItemFromMetaGraphDef()
/external/tensorflow/tensorflow/python/saved_model/
Dbuilder_impl.py35 from tensorflow.python.training import saver as tf_saver
296 saver = tf_saver.Saver(
310 meta_graph_def = saver.export_meta_graph(
384 saver = tf_saver.Saver(
394 saver.save(sess, variables_path, write_meta_graph=False, write_state=False)
406 meta_graph_def = saver.export_meta_graph(

123456