Searched refs:saver (Results 26 - 50 of 111) sorted by relevance

12345

/external/tensorflow/tensorflow/python/training/
H A Dsupervisor.py36 from tensorflow.python.training import saver as saver_mod
169 either the optionally `summary_op` and `saver` passed to the constructor, or
172 `None` to the `summary_op` and `saver` parameters.
202 # Value to pass for the 'ready_op', 'init_op', 'summary_op', 'saver',
219 saver=USE_DEFAULT,
269 saver: A Saver object. Used by chief supervisors if a `logdir` was
318 self._init_saver(saver=saver)
456 def _init_saver(self, saver=USE_DEFAULT):
457 """Initializes saver
599 def saver(self): member in class:Supervisor
[all...]
H A Devaluation_test.py41 from tensorflow.python.training import saver namespace
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)
H A Dsaver_test_utils.py25 from tensorflow.python.training import saver as saver_module
H A Dcheckpoint_utils_test.py33 from tensorflow.python.training import saver as saver_lib
46 saver = saver_lib.Saver()
47 saver.save(
67 saver = saver_lib.Saver()
68 saver.save(
/external/tensorflow/tensorflow/compiler/aot/tests/
H A 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)
75 saver_file = os.path.join(out_dir, 'test_graph_tfadd_with_ckpt_saver.saver')
77 f.write(saver.as_saver_def().SerializeToString())
/external/tensorflow/tensorflow/contrib/training/python/training/
H A Devaluation.py150 from tensorflow.python.training import saver as tf_saver
317 def _scaffold_with_init(scaffold, saver, checkpoint_path):
322 saver: The saver to use when restoring the checkpoint.
331 saver.restore(session, checkpoint_path)
341 saver=scaffold.saver)
H A 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),
[all...]
/external/tensorflow/tensorflow/contrib/rnn/python/tools/
H A Dcheckpoint_convert.py54 from tensorflow.python.training import saver as saver_lib
244 saver = saver_lib.Saver(new_variable_map, write_version=write_version)
249 saver.save(sess, checkpoint_to_path)
/external/tensorflow/tensorflow/contrib/session_bundle/
H A Dsession_bundle.py33 from tensorflow.python.training import saver as saver_lib
146 saver = saver_lib.import_meta_graph(meta_graph_def)
149 saver.restore(sess, os.path.join(export_dir, restore_files))
H A Dexporter.py38 from tensorflow.python.training import saver as tf_saver
138 saver: Saver object.
141 def __init__(self, saver):
142 # Makes a copy of the saver-def and disables garbage-collection, since the
144 # the exporter performs atomic copies of the saver output, it is required
145 # that garbage-collection via the underlying saver be disabled.
146 saver_def = saver.as_saver_def()
/external/tensorflow/tensorflow/examples/how_tos/reading_data/
H A Dfully_connected_preloaded.py81 # Create a saver for writing training checkpoints.
82 saver = tf.train.Saver()
124 saver.save(sess, FLAGS.train_dir, global_step=step)
129 saver.save(sess, FLAGS.train_dir, global_step=step)
H A Dfully_connected_preloaded_var.py87 # Create a saver for writing training checkpoints.
88 saver = tf.train.Saver()
135 saver.save(sess, FLAGS.train_dir, global_step=step)
140 saver.save(sess, FLAGS.train_dir, global_step=step)
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
H A Dexport_test.py39 from tensorflow.python.training import saver namespace
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/python/saved_model/
H A Dbuilder_impl.py35 from tensorflow.python.training import saver as tf_saver
294 # Initialize a saver to generate a sharded output for all saveables in the
296 saver = tf_saver.Saver(
310 meta_graph_def = saver.export_meta_graph(
382 # Initialize a saver to generate a sharded output for all saveables in the
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(
/external/tensorflow/tensorflow/contrib/slim/python/slim/
H A Dlearning.py269 from tensorflow.python.training import saver as tf_saver
550 saver=None,
604 saver: Saver to save checkpoints. If None, a default one will be created
638 if saver is not None:
639 raise ValueError('Cannot provide saver because logdir=None')
658 saver = saver or tf_saver.Saver()
733 saver=saver,
780 sv.saver
[all...]
H A 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/python/grappler/
H A Dcost_analyzer_tool.py35 from tensorflow.python.training import saver namespace
61 metagraph = saver.export_meta_graph(
/external/tensorflow/tensorflow/core/grappler/
H A Dgrappler_item_builder.cc44 #include "tensorflow/core/protobuf/saver.pb.h"
429 SaverDef saver; local
432 if (!saver.ParseFromString(raw)) {
435 if (saver.filename_tensor_name().empty()) {
438 new_item->save_op = saver.save_tensor_name();
439 new_item->restore_op = saver.restore_op_name();
440 new_item->save_restore_loc_tensor = saver.filename_tensor_name();
441 // Only use the first saver since it's not clear what to do if there's
446 const SaverDef& saver = meta_graph.saver_def(); local
447 new_item->save_op = saver
[all...]
/external/tensorflow/tensorflow/python/tools/
H A Dfreeze_graph.py57 from tensorflow.python.training import saver as saver_lib
104 saver = saver_lib.Saver(
106 saver.restore(sess, input_checkpoint)
129 saver = saver_lib.Saver(
131 saver.restore(sess, input_checkpoint)
199 print("Input saver file '" + input_saver + "' does not exist!")
289 help="TensorFlow saver file to load.")
H A Dfreeze_graph_test.py42 from tensorflow.python.training import saver as saver_lib
66 saver = saver_lib.Saver(write_version=saver_write_version)
67 checkpoint_path = saver.save(
180 saver = saver_lib.Saver()
181 checkpoint_path = saver.save(
/external/tensorflow/tensorflow/core/lib/io/
H A Dtable.cc130 void (*saver)(void*, const StringPiece&,
140 (*saver)(arg, block_iter->key(), block_iter->value());
129 InternalGet(const StringPiece& k, void* arg, void (*saver)(void*, const StringPiece&, const StringPiece&)) argument
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
H A Dquantile_ops.py35 from tensorflow.python.training import saver namespace
41 class QuantileAccumulator(saver.BaseSaverBuilder.SaveableObject):
93 return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix)
/external/tensorflow/tensorflow/contrib/framework/python/ops/
H A Dcheckpoint_ops_test.py34 from tensorflow.python.training import saver namespace
52 save = saver.Saver([bias])
111 save = saver.Saver([accum])
/external/tensorflow/tensorflow/python/keras/_impl/keras/
H A Destimator.py38 from tensorflow.python.training import saver as saver_lib
244 saver = saver_lib.Saver()
245 saver.save(sess, os.path.join(estimator.model_dir, 'keras_model.ckpt'))
/external/tensorflow/tensorflow/contrib/framework/python/framework/
H A Dcheckpoint_utils_test.py32 from tensorflow.python.training import saver as saver_lib
45 saver = saver_lib.Saver()
46 saver.save(
66 saver = saver_lib.Saver()
67 saver.save(

Completed in 609 milliseconds

12345