| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | saver_test.py | 22 from tensorflow.contrib.eager.python import saver as _saver
49 saver = _saver.Saver([v1])
50 saver.save(ckpt_prefix)
54 saver.restore(ckpt_prefix)
61 saver = _saver.Saver([v1, v2])
64 saver.save(ckpt_prefix)
71 saver = _saver.Saver([v1, v1])
73 saver.save(ckpt_prefix)
86 saver = _saver.Saver({'ckpt/v1': v1, 'ckpt/v2': v2})
87 saver.save(ckpt_prefix)
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | session_manager_test.py | 35 from tensorflow.python.training import saver as saver_lib
87 saver = saver_lib.Saver({"v": v})
91 saver=saver,
96 saver.save(sess, checkpoint_filename)
107 saver = saver_lib.Saver({"v": v})
114 saver=saver,
125 saver=saver,
144 saver = saver_lib.Saver({"v": v})
147 saver=saver,
171 saver = saver_lib.Saver({"v": v})
[all …]
|
| D | session_manager.py | 157 saver=None, argument
200 if not saver or not (checkpoint_dir or checkpoint_filename_with_path):
204 saver.restore(sess, checkpoint_filename_with_path)
220 saver.restore(sess, ckpt.model_checkpoint_path)
221 saver.recover_last_checkpoints(ckpt.all_model_checkpoint_paths)
227 saver=None, argument
285 saver,
318 saver=None, argument
350 saver,
|
| D | supervisor.py | 36 from tensorflow.python.training import saver as saver_mod
219 saver=USE_DEFAULT, argument
320 self._init_saver(saver=saver)
460 def _init_saver(self, saver=USE_DEFAULT): argument
467 if saver is Supervisor.USE_DEFAULT:
468 saver = self._get_first_op_from_collection(ops.GraphKeys.SAVERS)
469 if saver is None and variables.global_variables():
470 saver = saver_mod.Saver()
471 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
472 self._saver = saver
[all …]
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | builder_impl.py | 38 from tensorflow.python.training import saver as tf_saver
214 def _maybe_create_saver(self, saver=None): argument
216 if not saver:
219 saver = tf_saver.Saver(
224 return saver
233 saver=None): argument
280 saver = self._maybe_create_saver(saver)
289 meta_graph_def = saver.export_meta_graph(
307 saver=None): argument
361 saver = self._maybe_create_saver(saver)
[all …]
|
| D | load_v1_in_v2.py | 29 from tensorflow.python.training import saver as tf_saver
82 saver, _ = tf_saver._import_meta_graph_with_return_elements(
85 returns[0] = saver
87 def restore_variables(self, wrapped, saver): argument
89 if saver is not None:
90 saver_def = saver.saver_def
127 saver, = load_graph_returns
128 self.restore_variables(wrapped, saver)
|
| D | loader_impl.py | 36 from tensorflow.python.training import saver as tf_saver
352 def restore_variables(self, sess, saver, import_scope=None): argument
370 if (saver is None and
374 elif isinstance(saver, tf_saver.Saver):
375 saver.restore(sess, self._variables_path)
419 saver, _ = self.load_graph(sess.graph, tags, import_scope,
421 self.restore_variables(sess, saver, import_scope)
|
| D | loader_test.py | 38 from tensorflow.python.training import saver as tf_saver
138 saver, _ = loader.load_graph(
145 loader.restore_variables(sess, saver)
187 saver, _ = loader.load_graph(graph, ["foo_graph"])
189 loader.restore_variables(sess, saver)
244 saver=tf_saver.Saver([x, y]))
249 saver, _ = loader.load_graph(sess.graph, ["foo_graph"])
251 self.assertIsNotNone(saver)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| D | export.py | 39 from tensorflow.python.training import saver as tf_saver
56 saver = _get_first_op_from_collection(ops.GraphKeys.SAVERS)
57 if saver is not None:
58 if saver:
59 saver = saver[0]
61 saver = None
62 if saver is None and variables.global_variables():
63 saver = tf_saver.Saver()
64 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
65 return saver
[all …]
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| D | evaluation.py | 133 from tensorflow.python.training import saver as tf_saver
203 saver = None
205 saver = tf_saver.Saver(variables_to_restore)
211 init_op=initial_op, init_feed_dict=initial_op_feed_dict, saver=saver),
292 saver = None
294 saver = tf_saver.Saver(variables_to_restore)
301 init_fn=init_fn, saver=saver),
|
| D | learning.py | 269 from tensorflow.python.training import saver as tf_saver
550 saver=None, argument
639 if saver is not None:
659 saver = saver or tf_saver.Saver()
734 saver=saver,
781 sv.saver.save(sess, sv.save_path, global_step=sv.global_step)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | dataset_serialization_test_base.py | 37 from tensorflow.python.training import saver as saver_lib
267 saver = self._import_meta_graph()
272 self._restore(saver, sess)
331 _, get_next_op, saver = self._build_graph(
335 self._restore(saver, sess)
392 get_next_op, saver = self._build_empty_graph(
396 self._restore(saver, sess)
426 init_op, get_next_op, saver = self._build_graph(
434 self._save(sess, saver)
523 saver = self._import_meta_graph()
[all …]
|
| D | serialization_integration_test.py | 26 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/checkpoint/python/ |
| D | python_state_test.py | 39 saver = util.Checkpoint(numpy=save_state)
47 first_save_path = saver.save(prefix)
50 second_save_path = saver.save(prefix)
73 saver = util.Checkpoint(numpy=save_state)
75 save_path = saver.save(prefix)
76 saver.restore(save_path)
78 saver.save(prefix)
80 saver.save(prefix)
81 saver.restore(save_path)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | graph_actions.py | 52 from tensorflow.python.training import saver as tf_saver
101 def _restore_from_checkpoint(session, graph, checkpoint_path, saver=None): argument
103 saver = saver or _make_saver(graph)
104 if saver:
105 saver.restore(session, checkpoint_path)
308 saver=_make_saver(graph, keep_checkpoint_max),
389 supervisor.saver.save(session, ckpt_path, global_step=last_step)
422 saver = _get_first_op_from_collection(ops.GraphKeys.SAVERS)
423 if saver is None and variables.global_variables():
424 saver = tf_saver.Saver()
[all …]
|
| /external/tensorflow/tensorflow/contrib/rnn/python/tools/ |
| D | checkpoint_convert_test.py | 29 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/python/training/tracking/ |
| D | python_state_test.py | 153 saver = util.Checkpoint(numpy=save_state)
161 first_save_path = saver.save(prefix)
164 second_save_path = saver.save(prefix)
187 saver = util.Checkpoint(numpy=save_state)
189 save_path = saver.save(prefix)
190 saver.restore(save_path)
192 saver.save(prefix)
194 saver.save(prefix)
195 saver.restore(save_path)
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | bidirectional_sequence_rnn_test.py | 153 saver, argument
157 saver.save(sess, model_dir)
165 saver = tf.train.Saver()
166 saver.restore(new_sess, model_dir)
218 saver = tf.train.Saver()
220 self.buildRnnLayer(), self.buildRnnLayer(), sess, saver, False)
239 saver = tf.train.Saver()
244 saver,
262 saver = tf.train.Saver()
267 saver,
[all …]
|
| D | unidirectional_sequence_rnn_test.py | 116 def saveAndRestoreModel(self, rnn_layer, sess, saver, is_dynamic_rnn): argument
135 saver.save(sess, model_dir)
142 saver = tf.train.Saver()
143 saver.restore(new_sess, model_dir)
190 saver = tf.train.Saver()
192 self.buildRnnLayer(), sess, saver, is_dynamic_rnn=False)
208 saver = tf.train.Saver()
211 self.buildRnnLayer(), sess, saver, is_dynamic_rnn=True)
|
| D | unidirectional_sequence_lstm_test.py | 121 def saveAndRestoreModel(self, lstm_layer, sess, saver, is_dynamic_rnn): argument
123 saver.save(sess, model_dir)
130 saver = tf.train.Saver()
131 saver.restore(new_sess, model_dir)
182 saver = tf.train.Saver()
184 self.buildLstmLayer(), sess, saver, is_dynamic_rnn=False)
200 saver = tf.train.Saver()
203 self.buildLstmLayer(), sess, saver, is_dynamic_rnn=True)
|
| D | bidirectional_sequence_lstm_test.py | 130 def saveAndRestoreModel(self, fw_lstm_layer, bw_lstm_layer, sess, saver, argument
133 saver.save(sess, model_dir)
141 saver = tf.train.Saver()
142 saver.restore(new_sess, model_dir)
197 saver = tf.train.Saver()
199 self.buildLstmLayer(), self.buildLstmLayer(), sess, saver, False)
215 saver = tf.train.Saver()
220 saver,
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver_test.py | 35 saver = functional_saver.Saver(
38 save_path = saver.save(constant_op.constant(prefix))
40 saver.restore(save_path)
51 saver = functional_saver.Saver(
57 lambda: proto_accumulator.append(saver.to_proto()), signature=())
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | stats_accumulator_ops.py | 28 from tensorflow.python.training import saver
35 class StatsAccumulatorSaveable(saver.BaseSaverBuilder.SaveableObject):
47 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
49 saver.BaseSaverBuilder.SaveSpec(num_updates, slice_spec,
51 saver.BaseSaverBuilder.SaveSpec(partition_ids, slice_spec,
53 saver.BaseSaverBuilder.SaveSpec(feature_ids, slice_spec,
55 saver.BaseSaverBuilder.SaveSpec(gradients, slice_spec,
57 saver.BaseSaverBuilder.SaveSpec(hessians, slice_spec,
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | training_test.py | 42 from tensorflow.python.training import saver as saver_lib
334 saver = saver_lib.Saver()
343 logdir, save_steps=50, saver=saver),
387 saver = saver_lib.Saver()
393 logdir1, save_steps=1, saver=saver),
404 saver = saver_lib.Saver()
410 logdir1, save_steps=300, saver=saver),
467 saver = saver_lib.Saver()
473 logdir, save_steps=200, saver=saver),
491 saver = saver_lib.Saver()
[all …]
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | values_test.py | 38 from tensorflow.python.training import saver as saver_lib
383 saver = saver_lib.Saver(var_list=[var])
386 return saver.save(sess, prefix), saver
405 save_path, saver = self._save_return_saver(sess, mirrored)
411 saver.restore(sess, save_path)
456 saver = saver_lib.Saver(var_list=[var])
457 saver.restore(sess, save_path)
470 saver = saver_lib.Saver(var_list=[mirrored])
471 saver.restore(sess, save_path)
616 saver = saver_lib.Saver(var_list=[var])
[all …]
|