Home
last modified time | relevance | path

Searched refs:saver_lib (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/python/training/
Dsession_manager_test.py31 from tensorflow.python.training import saver as saver_lib unknown
99 saver = saver_lib.Saver({"v": v})
119 saver = saver_lib.Saver({"v": v})
156 saver = saver_lib.Saver({"v": v})
182 saver = saver_lib.Saver({"v": v})
237 saver = saver_lib.Saver({"v": v})
261 saver = saver_lib.Saver({"v": v})
295 saver = saver_lib.Saver({"v": v})
318 saver = saver_lib.Saver({"v": v})
386 saver = saver_lib.Saver({"v": v})
[all …]
Dcheckpoint_utils_test.py38 from tensorflow.python.training import saver as saver_lib unknown
51 saver = saver_lib.Saver()
72 saver = saver_lib.Saver()
479 saver = saver_lib.Saver(sharded=True)
Dwarm_starting_util.py27 from tensorflow.python.training import saver as saver_lib unknown
389 names_to_keys = saver_lib.object_graph_key_mapping(fname)
Dcheckpoint_ops_test.py31 from tensorflow.python.training import saver as saver_lib unknown
51 saver = saver_lib.Saver()
Dmonitored_session_test.py49 from tensorflow.python.training import saver as saver_lib unknown
90 self.assertTrue(isinstance(scaffold.saver, saver_lib.Saver))
113 self.assertTrue(isinstance(scaffold.saver, saver_lib.Saver))
132 ops.add_to_collection(ops.GraphKeys.SAVERS, saver_lib.Saver())
133 ops.add_to_collection(ops.GraphKeys.SAVERS, saver_lib.Saver())
140 saver = saver_lib.Saver()
172 saver = saver_lib.Saver()
197 saver = saver_lib.Saver()
1529 save_path = saver_lib._get_saver_or_default().save(
1562 save_path = saver_lib._get_saver_or_default().save(
Dsupervisor_test.py45 from tensorflow.python.training import saver as saver_lib unknown
748 new_saver = saver_lib.import_meta_graph(".".join([filename, "meta"]))
804 sav = saver_lib.Saver([v])
869 sav = saver_lib.Saver([v])
Dmoving_averages_test.py31 from tensorflow.python.training import saver as saver_lib unknown
552 meta_graph = saver_lib.export_meta_graph(
556 _ = saver_lib.import_meta_graph(meta_graph)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dcheckpoint_input_pipeline_hook_test.py29 from tensorflow.python.training import saver as saver_lib unknown
61 saver_lib.import_meta_graph(meta_filename)
62 saver = saver_lib.Saver()
Dmake_saveable_from_iterator_test.py26 from tensorflow.python.training import saver as saver_lib unknown
48 saver = saver_lib.Saver()
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_values_test.py35 from tensorflow.python.training import saver as saver_lib unknown
114 saver = saver_lib.Saver(var_list=[var])
164 saver = saver_lib.Saver(var_list=[var])
178 saver = saver_lib.Saver(var_list=[mirrored])
Dwarm_starting_util_test.py34 from tensorflow.python.training import saver as saver_lib unknown
65 saver = saver_lib.Saver()
Dcheckpoint_utils_test.py35 from tensorflow.python.training import saver as saver_lib unknown
45 saver = saver_lib.Saver()
Dvalues_test.py45 from tensorflow.python.training import saver as saver_lib unknown
449 saver = saver_lib.Saver(var_list=[var])
708 saver = saver_lib.Saver(var_list=[var])
723 saver = saver_lib.Saver(var_list=[replica_local])
738 saver = saver_lib.Saver(var_list=[replica_local])
/external/tensorflow/tensorflow/python/tools/
Dfreeze_graph.py55 from tensorflow.python.training import saver as saver_lib unknown
145 saver = saver_lib.Saver(
149 restorer = saver_lib.import_meta_graph(
185 saver = saver_lib.Saver(
Dfreeze_graph_test.py44 from tensorflow.python.training import saver as saver_lib unknown
66 saver = saver_lib.Saver(write_version=saver_write_version)
182 saver = saver_lib.Saver()
305 saver = saver_lib.Saver()
Dsaved_model_aot_compile.py37 from tensorflow.python.training import saver as saver_lib unknown
280 restorer = saver_lib.import_meta_graph(meta_graph_def, clear_devices=True)
/external/tensorflow/tensorflow/python/data/experimental/ops/
Diterator_ops.py22 from tensorflow.python.training import saver as saver_lib unknown
299 class _CustomSaver(saver_lib.Saver):
/external/tensorflow/tensorflow/compiler/aot/tests/
Dmake_test_graphs.py38 from tensorflow.python.training import saver as saver_lib unknown
55 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
70 saver = saver_lib.Saver(name='abcprefix', write_version=saver_pb2.SaverDef.V1)
/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint.py33 from tensorflow.python.training import saver as saver_lib unknown
46 saver: Optional[saver_lib.Saver] = None,
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental.py37 from tensorflow.python.training import saver as saver_lib unknown
318 saver=saver_lib.Saver(
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test_base.py36 from tensorflow.python.training import saver as saver_lib unknown
524 saver = saver_lib.Saver(allow_empty=True)
606 return saver_lib.import_meta_graph(meta_file_path)
Dshuffle_test.py39 from tensorflow.python.training import saver as saver_lib unknown
490 saver = saver_lib.Saver(allow_empty=True)
/external/tensorflow/tensorflow/python/kernel_tests/variables/
Dpartitioned_variables_test.py32 from tensorflow.python.training import saver as saver_lib unknown
616 saver = saver_lib.Saver()
625 saver = saver_lib.import_meta_graph(save_path + ".meta")
/external/tensorflow/tensorflow/python/eager/
Dwrap_function_test.py38 from tensorflow.python.training import saver as saver_lib unknown
124 saver = saver_lib.Saver([v])
130 saver_lib.import_meta_graph(save_prefix + '.meta')
/external/tensorflow/tensorflow/python/checkpoint/
Dcheckpoint_test.py48 from tensorflow.python.training import saver as saver_lib unknown
146 class _MirroringSaveable(saver_lib.BaseSaverBuilder.SaveableObject):
152 spec = saver_lib.BaseSaverBuilder.SaveSpec(
224 saver = saver_lib.Saver(var_list=[v])

12