| /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
53 saver = _saver.Saver([v1])
54 saver.save(ckpt_prefix)
58 saver.restore(ckpt_prefix)
71 saver = _saver.Saver([v1_first_graph, v1_second_graph])
74 saver.save(ckpt_prefix)
81 saver = _saver.Saver([v1, v1]
[all...] |
| saver.py | 0 """Saver for eager mode TensorFlow."""
27 from tensorflow.python.training import saver as _saver
116 class Saver(object):
117 """A tf.train.Saver adapter for use when eager execution is enabled.
121 """A tf.train.Saver adapter for use when eager execution is enabled.
123 The API, and on-disk format, mimic tf.train.Saver except that no
135 raise RuntimeError("tfe.Saver can only be used when eager "
136 "execution is enabled. Use tf.train.Saver when "
138 self._saver = _saver.Saver(var_list=var_list)
|
| tfe.py | 48 @@Saver
82 from tensorflow.contrib.eager.python.saver import get_optimizer_variables
83 from tensorflow.contrib.eager.python.saver import restore_variables_on_create
84 from tensorflow.contrib.eager.python.saver import Saver
|
| /external/tensorflow/tensorflow/python/training/ |
| saver_test.py | 15 """Tests for tensorflow.python.training.saver.py."""
71 from tensorflow.python.training import saver as saver_module
102 save = saver_module.Saver(
126 save = saver_module.Saver({"v0": v0, "v1": v1, "v2": v2.saveable})
154 save2 = saver_module.Saver({"v0": v0_2, "v1": v1_2, "v2": v2_2.saveable})
176 saver_module.Saver()
187 save = saver_module.Saver([v1, v2])
207 graph_saver = saver_module.Saver([w1, w2])
218 graph_saver = saver_module.Saver([w1, w2])
233 graph_saver = saver_module.Saver([w3, w4]
[all...] |
| saver_large_partitioned_variable_test.py | 15 """Tests for tensorflow.python.training.saver.py."""
30 from tensorflow.python.training import saver
50 save = saver.Saver(partitioned_var)
|
| 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_test.py | 33 from tensorflow.python.training import saver as saver_lib
84 saver = saver_lib.Saver({"v": v})
88 saver=saver,
93 saver.save(sess, checkpoint_filename)
104 saver = saver_lib.Saver({"v": v})
111 saver=saver,
[all...] |
| /external/llvm/lib/LibDriver/ |
| LibDriver.cpp | 69 StringSaver &Saver) {
82 StringRef Env = Saver.save(*EnvOpt);
105 StringSaver Saver(Alloc);
106 cl::ExpandResponseFiles(Saver, cl::TokenizeWindowsCommandLine, NewArgs);
129 std::vector<StringRef> SearchPaths = getSearchPaths(&Args, Saver);
139 NewArchiveMember::getFile(Saver.save(*Path), /*Deterministic=*/true);
|
| /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...] |
| 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/clang/tools/driver/ |
| driver.cpp | 224 static void getCLEnvVarOptions(std::string &EnvValue, llvm::StringSaver &Saver,
226 llvm::cl::TokenizeWindowsCommandLine(EnvValue, Saver, Opts);
333 llvm::StringSaver Saver(A);
370 llvm::cl::ExpandResponseFiles(Saver, Tokenizer, argv, MarkEOLs);
403 getCLEnvVarOptions(OptCL.getValue(), Saver, PrependedOpts);
412 getCLEnvVarOptions(Opt_CL_.getValue(), Saver, AppendedOpts);
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| evaluation.py | 128 from tensorflow.python.training import saver as tf_saver
198 saver = None
200 saver = tf_saver.Saver(variables_to_restore)
206 init_op=initial_op, init_feed_dict=initial_op_feed_dict, saver=saver),
283 saver = None
285 saver = tf_saver.Saver(variables_to_restore)
292 init_fn=init_fn, saver=saver)
[all...] |
| 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
[all...] |
| /build/kati/ |
| depgraph.go | 223 // Saver is the interface that saves DepGraph.
224 type Saver interface {
231 Saver
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| 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/framework/python/ops/ |
| checkpoint_ops_test.py | 34 from tensorflow.python.training import saver
52 save = saver.Saver([bias])
111 save = saver.Saver([accum])
|
| /external/swiftshader/third_party/llvm-subzero/lib/Support/ |
| CommandLine.cpp | 705 void cl::TokenizeGNUCommandLine(StringRef Src, StringSaver &Saver,
747 NewArgv.push_back(Saver.save(StringRef(Token)).data());
758 NewArgv.push_back(Saver.save(StringRef(Token)).data());
802 void cl::TokenizeWindowsCommandLine(StringRef Src, StringSaver &Saver,
839 NewArgv.push_back(Saver.save(StringRef(Token)).data());
874 NewArgv.push_back(Saver.save(StringRef(Token)).data());
886 static bool ExpandResponseFile(StringRef FName, StringSaver &Saver,
912 Tokenizer(Str, Saver, NewArgv, MarkEOLs);
933 NewArgv[I] = Saver.save(ResponseFile.c_str()).data();
943 bool cl::ExpandResponseFiles(StringSaver &Saver, TokenizerCallback Tokenizer
[all...] |
| /external/llvm/lib/Support/ |
| CommandLine.cpp | 686 void cl::TokenizeGNUCommandLine(StringRef Src, StringSaver &Saver,
728 NewArgv.push_back(Saver.save(Token.c_str()));
739 NewArgv.push_back(Saver.save(Token.c_str()));
783 void cl::TokenizeWindowsCommandLine(StringRef Src, StringSaver &Saver,
820 NewArgv.push_back(Saver.save(Token.c_str()));
855 NewArgv.push_back(Saver.save(Token.c_str()));
867 static bool ExpandResponseFile(const char *FName, StringSaver &Saver,
893 Tokenizer(Str, Saver, NewArgv, MarkEOLs);
900 bool cl::ExpandResponseFiles(StringSaver &Saver, TokenizerCallback Tokenizer,
929 if (!ExpandResponseFile(Arg + 1, Saver, Tokenizer, ExpandedArgv
[all...] |
| /external/llvm/unittests/Support/ |
| CommandLineTest.cpp | 167 typedef void ParserFunction(StringRef Source, StringSaver &Saver,
175 StringSaver Saver(A);
176 parse(Input, Saver, Actual, /*MarkEOLs=*/false);
|
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| 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/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)
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| moving_average_optimizer.py | 28 from tensorflow.python.training import saver
53 // Then create your saver like this:
54 saver = opt.swapping_saver()
59 saver=saver)
62 Note that for evaluation, the normal saver should be used instead of
110 """Create a saver swapping moving averages and variables.
112 You should use this saver during training. It will save the moving averages
114 evaluations or inference you should use a regular saver and it will
121 var_list: List of variables to save, as per `Saver()`
[all...] |
| /external/tensorflow/tensorflow/contrib/session_bundle/example/ |
| export_half_plus_two.py | 62 # Setup a standard Saver for our variables.
63 save = tf.train.Saver(
|
| /external/tensorflow/tensorflow/examples/how_tos/reading_data/ |
| 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)
|
| 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)
|
