| /external/tensorflow/tensorflow/contrib/eager/python/ |
| saver_test.py | 1 """Tests for eager mode Saver."""
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]
[all...] |
| /external/tensorflow/tensorflow/contrib/checkpoint/python/ |
| 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/slim/python/slim/ |
| 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)
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| test_util.py | 23 from tensorflow.python.training import saver
46 metagraph = saver.export_meta_graph(graph_def=graph.as_graph_def())
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| functional_saver_test.py | 15 """Tests for the functional saver."""
35 saver = functional_saver.Saver(
38 save_path = saver.save(constant_op.constant(prefix))
40 saver.restore(save_path)
44 second_saver = functional_saver.Saver(
51 saver = functional_saver.Saver(
57 lambda: proto_accumulator.append(saver.to_proto()), signature=())
72 second_saver = functional_saver.Saver(
[all...] |
| /external/tensorflow/tensorflow/contrib/rnn/python/tools/ |
| 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
[all...] |
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| 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())
|
| 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)
492 Saver checkpoint
[all...] |
| shuffle_dataset_serialization_test.py | 25 from tensorflow.python.training import saver as saver_lib
138 saver = saver_lib.Saver(allow_empty=True)
140 self._save(sess, saver)
142 self._restore(saver, sess)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| export.py | 39 from tensorflow.python.training import saver as tf_saver
55 """Lazy init and return 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(
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| 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,
[all...] |
| saver_large_variable_test.py | 15 """Tests for tensorflow.python.training.saver.py."""
31 from tensorflow.python.training import saver
49 save = saver.Saver(
|
| session_manager.py | 68 sess = sm.prepare_session(master, init_op, saver, checkpoint_dir)
75 and `saver` as an argument.
157 saver=None,
168 saver: A `Saver` object used to restore a model.
198 # If either saver or checkpoint_* is not specified, cannot restore. Just
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
[all...] |
| quantize_training_test.py | 32 from tensorflow.python.training import saver as saver_module
69 saver = saver_module.Saver({'b': b})
84 saver.save(sess, save_path)
91 saver.restore(sess, save_path)
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| learning.py | 81 saver=None,
136 saver: Saver to save checkpoints. If None, a default one will be created
185 saver=saver,
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| 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/python/saved_model/ |
| builder_impl.py | 38 from tensorflow.python.training import saver as tf_saver
214 def _maybe_create_saver(self, saver=None):
215 """Creates a sharded saver if one does not already exist."""
216 if not saver:
217 # Initialize a saver to generate a sharded output for all saveables in the
219 saver = tf_saver.Saver(
224 return saver
233 saver=None):
236 Creates a Saver in the current scope and uses the Saver to export the met
[all...] |
| 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):
89 if saver is not None:
90 saver_def = saver.saver_def
127 saver, = load_graph_returns
128 self.restore_variables(wrapped, saver)
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| warm_starting_util_test.py | 36 from tensorflow.python.training import saver as saver_lib
64 saver = saver_lib.Saver()
66 saver.save(sess, ckpt_prefix, global_step=0)
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| bidirectional_sequence_rnn_test.py | 153 saver,
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()
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| tensor_forest_ops.py | 23 from tensorflow.python.training import saver
26 class TreeVariableSaveable(saver.BaseSaverBuilder.SaveableObject):
44 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name)]
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| graph_actions.py | 52 from tensorflow.python.training import saver as tf_saver
94 return tf_saver.Saver(vars_to_save,
101 def _restore_from_checkpoint(session, graph, checkpoint_path, saver=None):
103 saver = saver or _make_saver(graph)
104 if saver:
105 saver.restore(session, checkpoint_path)
107 logging.info('No variables found in graph, not creating Saver() object.')
186 arg to tf.train.Saver constructor.
308 saver=_make_saver(graph, keep_checkpoint_max)
[all...] |
| /device/generic/goldfish-opengl/android-emu/android/base/files/ |
| StreamSerializing.h | 65 void saveBuffer(Stream* stream, const std::vector<T>& buffer, SaveFunc&& saver) {
68 saver(stream, val);
95 void saveCollection(Stream* stream, const Collection& c, SaveFunc&& saver) {
98 saver(stream, val);
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| model_ops.py | 35 from tensorflow.python.training import saver
43 class TreeEnsembleVariableSavable(saver.BaseSaverBuilder.SaveableObject):
60 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
62 saver.BaseSaverBuilder.SaveSpec(ensemble_config, slice_spec,
|
| /external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/ |
| cifar10_eval.py | 50 def eval_once(saver, summary_writer, top_k_op, summary_op):
54 saver: Saver.
63 saver.restore(sess, ckpt.model_checkpoint_path)
122 saver = tf.train.Saver(variables_to_restore)
130 eval_once(saver, summary_writer, top_k_op, summary_op)
|
