/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)
|