Home
last modified time | relevance | path

Searched refs:save_path (Results 1 – 25 of 59) sorted by relevance

123

/external/tensorflow/tensorflow/contrib/distribute/python/
Dvalues_test.py389 save_path, _ = self._save_return_saver(sess, var)
390 return save_path
405 save_path, saver = self._save_return_saver(sess, mirrored)
411 saver.restore(sess, save_path)
424 save_path = self._save(sess, mirrored)
428 return save_path
440 save_path = self._save(sess, var)
444 return save_path
446 def _restore_normal(self, save_path): argument
457 saver.restore(sess, save_path)
[all …]
/external/tensorflow/tensorflow/python/training/tracking/
Dutil.py59 def __init__(self, object_graph_proto, save_path, save_path_tensor, argument
95 self.save_path_string = save_path
97 save_path).get_variable_to_dtype_map()
174 def __init__(self, save_path, dtype_map=None): argument
175 self.save_path = save_path
235 prefix=self.save_path,
299 def object_metadata(save_path): argument
321 reader = pywrap_tensorflow.NewCheckpointReader(save_path)
330 save_path, base.OBJECT_GRAPH_PROTO_KEY))
815 sess=session, save_path=self._checkpoint.save_path)
[all …]
Dutil_test.py181 save_path = checkpoint.save(checkpoint_prefix)
183 objects = trackable_utils.object_metadata(save_path)
368 save_path = checkpoint.save(prefix)
371 checkpoint.restore(save_path).assert_consumed().initialize_or_restore()
375 save_path = checkpoint.save(prefix)
377 checkpoint.restore(save_path).assert_consumed().initialize_or_restore()
390 save_path = saver.save(sess, prefix)
393 saver.restore(sess, save_path)
417 save_path = root_trackable.save(file_prefix=prefix)
424 status = root_trackable.restore(save_path=save_path).assert_consumed()
[all …]
Ddata_structures_test.py97 save_path = os.path.join(self.get_temp_dir(), "ckpt")
98 model.save_weights(save_path)
100 model.load_weights(save_path)
442 save_path = os.path.join(self.get_temp_dir(), "ckpt")
443 model.save_weights(save_path)
445 model.load_weights(save_path)
501 save_path = os.path.join(self.get_temp_dir(), "ckpt")
503 model.save_weights(save_path)
512 save_path = os.path.join(self.get_temp_dir(), "ckpt")
513 model.save_weights(save_path)
[all …]
Dutil_with_v1_optimizers_test.py210 save_path = root_trackable.save(file_prefix=prefix)
216 status = root_trackable.restore(save_path=save_path).assert_consumed()
232 status = on_create_root.restore(save_path=save_path)
368 status = root.restore(save_path=checkpoint_path)
402 status = root.restore(save_path=manager.latest_checkpoint)
435 status = root.restore(save_path=checkpoint_path)
593 save_path = obj.save(checkpoint_prefix)
594 obj.restore(save_path)
596 obj.restore(save_path)
635 save_path = second_root_trackable.save(checkpoint_prefix)
[all …]
Dpython_state_test.py142 save_path = root.save(prefix)
145 root.restore(save_path)
189 save_path = saver.save(prefix)
190 saver.restore(save_path)
195 saver.restore(save_path)
202 save_path = checkpoint.save(os.path.join(self.get_temp_dir(), "ckpt"))
204 checkpoint.restore(save_path)
208 second_checkpoint.restore(save_path)
Dbase_test.py67 save_path = saved.save(os.path.join(self.get_temp_dir(), "ckpt"))
69 restored.restore(save_path).assert_consumed()
80 save_path = saved.save(os.path.join(self.get_temp_dir(), "ckpt"))
82 status = restored.restore(save_path)
/external/tensorflow/tensorflow/contrib/eager/python/
Dnetwork.py756 network, save_path, global_step=None, map_func=None): argument
788 if os.path.isdir(save_path):
791 save_path = os.path.join(save_path, network.name.replace("/", "_"))
829 sess=sess, save_path=save_path, write_meta_graph=False,
866 def _restore_existing_variables(network, save_path, map_func, user_map_func): argument
910 sess=sess, save_path=save_path)
914 def _set_restore_on_create(network, save_path, map_func, user_map_func, argument
917 checkpoint_reader = checkpoint_utils.load_checkpoint(save_path)
919 for checkpoint_name, _ in checkpoint_utils.list_variables(save_path):
957 def restore_network_checkpoint(network, save_path, map_func=None): argument
[all …]
Dsaver.py48 def restore_variables_on_create(save_path, map_func=None): argument
80 if save_path:
89 reader = checkpoint_utils.load_checkpoint(save_path)
90 for k, _ in checkpoint_utils.list_variables(save_path):
108 if save_path:
/external/tensorflow/tensorflow/python/training/saving/
Dfunctional_saver_test.py38 save_path = saver.save(constant_op.constant(prefix))
40 saver.restore(save_path)
46 second_saver.restore(save_path)
66 save_path = save(constant_op.constant(prefix))
68 restore(constant_op.constant(save_path))
74 second_saver.restore(save_path)
/external/tensorflow/tensorflow/python/training/
Dsaver_test.py85 save_path = os.path.join(self.get_temp_dir(), "basic_save_restore")
112 val = save.save(sess, save_path)
114 self.assertEqual(save_path, val)
131 save.restore(sess, save_path)
159 save2.restore(sess, save_path)
271 save_path = os.path.join(self.get_temp_dir(), "resource_cache")
280 save.save(sess, save_path)
283 save2.restore(sess, save_path)
395 save_path = os.path.join(self.get_temp_dir(), "int64")
404 val = save.save(sess, save_path)
[all …]
Dsaver.py1074 save_path, argument
1148 checkpoint_file = "%s-%d" % (save_path, global_step)
1151 checkpoint_file = "%s-%s" % (save_path, "{:08d}".format(global_step))
1153 checkpoint_file = save_path
1155 save_path) == latest_filename and not self._sharded:
1159 (latest_filename, save_path))
1165 save_path_parent = os.path.dirname(save_path)
1191 save_path))
1253 def restore(self, sess, save_path): argument
1273 if save_path is None:
[all …]
Dsaver_large_partitioned_variable_test.py39 save_path = os.path.join(self.get_temp_dir(), "large_variable")
56 val = save.save(sess, save_path)
57 self.assertEqual(save_path, val)
Dquantize_training_test.py58 save_path = os.path.join(self.get_temp_dir(), 'quantized_save_restore')
84 saver.save(sess, save_path)
91 saver.restore(sess, save_path)
Dsupervisor_test.py436 self._wait_for_glob(sv.save_path, 3.0)
605 save_path = sv.save_path
606 self._wait_for_glob(save_path, 3.0)
751 filename = sv.saver.save(sess, sv.save_path)
760 sv2.saver.save(sess, sv2.save_path)
777 save_path = sv.save_path
778 self._wait_for_glob(save_path, 3.0)
812 sav.restore(sess, save_path)
828 save_path = "%s-123" % sv.save_path
829 self._wait_for_glob(save_path, 3.0)
[all …]
Dcheckpoint_management.py694 save_path = self._checkpoint.write(prefix)
698 if save_path in self._maybe_delete:
699 del self._maybe_delete[save_path]
700 self._maybe_delete[save_path] = timestamp
701 self._latest_checkpoint = save_path
704 return save_path
/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_checkpoint_test.py48 save_path = checkpoint.save(checkpoint_prefix)
51 checkpoint.restore(save_path).run_restore_ops()
82 save_path = checkpoint.save(checkpoint_prefix)
86 checkpoint.restore(save_path).run_restore_ops()
102 save_path = checkpoint.save(checkpoint_prefix)
104 checkpoint.restore(save_path).run_restore_ops()
106 save_path = checkpoint.save(checkpoint_prefix)
107 checkpoint.restore(save_path).run_restore_ops()
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dmoving_average_optimizer_test.py115 save_path = os.path.join(save_dir, 'model')
159 train_saver.save(sess, save_path)
160 train_saver.restore(sess, save_path)
168 subset_save_path = save_path + '_subset'
182 train_saver.restore(sess, save_path)
193 train_saver.save(sess, save_path)
194 inference_saver.restore(sess, save_path)
202 train_saver.restore(sess, save_path)
213 train_saver.save(sess, save_path)
214 inference_saver.restore(sess, save_path)
/external/tensorflow/tensorflow/contrib/checkpoint/python/
Dpython_state_test.py75 save_path = saver.save(prefix)
76 saver.restore(save_path)
81 saver.restore(save_path)
95 save_path = checkpoint.save(os.path.join(self.get_temp_dir(), "ckpt"))
97 checkpoint.restore(save_path)
101 second_checkpoint.restore(save_path)
Dsplit_dependency_test.py84 save_path = save_checkpoint.save(checkpoint_prefix)
90 save_path).assert_consumed().run_restore_ops()
96 status = one_dep_restore_checkpoint.restore(save_path)
104 status = restore_checkpoint.restore(save_path)
Dvisualize.py25 def dot_graph_from_checkpoint(save_path): argument
53 reader = pywrap_tensorflow.NewCheckpointReader(save_path)
54 object_graph = trackable_utils.object_metadata(save_path)
Dcontainers_test.py53 save_path = save_root.save(checkpoint_prefix)
58 status = restore_root.restore(save_path)
90 save_path = checkpoint.save(checkpoint_prefix)
91 metadata = util.object_metadata(save_path)
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Dcheckpointable_utils_test.py228 save_path = root_trackable.save(file_prefix=prefix)
234 status = root_trackable.restore(save_path=save_path).assert_consumed()
251 status = on_create_root.restore(save_path=save_path)
315 status = root.restore(save_path=checkpoint_path)
348 status = root.restore(save_path=checkpoint_path)
382 status = root.restore(save_path=checkpoint_path)
529 save_path = obj.save(checkpoint_prefix)
530 obj.restore(save_path)
532 obj.restore(save_path)
571 save_path = second_root_trackable.save(checkpoint_prefix)
[all …]
/external/fonttools/Tests/cffLib/
DcffLib_test.py53 save_path = temp_path[:-4] + '2.otf'
59 font.save(save_path)
60 font2 = TTFont(save_path)
/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/
Dquantile_ops_test.py146 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
165 save.save(sess, save_path)
171 save.restore(sess, save_path)
178 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
193 save.save(sess, save_path)
202 save.restore(sess, save_path)

123