/external/tensorflow/tensorflow/python/training/ |
H A D | supervisor.py | 36 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 D | evaluation_test.py | 41 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 D | saver_test_utils.py | 25 from tensorflow.python.training import saver as saver_module
|
H A D | checkpoint_utils_test.py | 33 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 D | make_test_graphs.py | 35 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 D | evaluation.py | 150 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 D | training_test.py | 41 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 D | checkpoint_convert.py | 54 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 D | session_bundle.py | 33 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 D | exporter.py | 38 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 D | fully_connected_preloaded.py | 81 # 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 D | fully_connected_preloaded_var.py | 87 # 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 D | export_test.py | 39 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 D | builder_impl.py | 35 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 D | learning.py | 269 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 D | evaluation_test.py | 46 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 D | cost_analyzer_tool.py | 35 from tensorflow.python.training import saver namespace 61 metagraph = saver.export_meta_graph(
|
/external/tensorflow/tensorflow/core/grappler/ |
H A D | grappler_item_builder.cc | 44 #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 D | freeze_graph.py | 57 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 D | freeze_graph_test.py | 42 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 D | table.cc | 130 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 D | quantile_ops.py | 35 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 D | checkpoint_ops_test.py | 34 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 D | estimator.py | 38 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 D | checkpoint_utils_test.py | 32 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(
|