| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util_test.py | 193 save_path = checkpoint.save(checkpoint_prefix)
195 objects = trackable_utils.object_metadata(save_path)
371 save_path = checkpoint.save(prefix)
374 checkpoint.restore(save_path).assert_consumed().initialize_or_restore()
378 save_path = checkpoint.save(prefix)
380 checkpoint.restore(save_path).assert_consumed().initialize_or_restore()
393 save_path = saver.save(sess, prefix)
396 saver.restore(sess, save_path)
407 save_path = ckpt.save(file_prefix=prefix)
408 status = ckpt.restore(save_path=save_path)
[all …]
|
| D | util.py | 165 def __init__(self, object_graph_proto, save_path, save_path_tensor, argument
202 self.save_path_string = save_path
204 save_path).get_variable_to_dtype_map()
301 def __init__(self, save_path, dtype_map=None): argument
302 self.save_path = save_path
363 prefix=self.save_path,
442 def object_metadata(save_path): argument
465 reader = py_checkpoint_reader.NewCheckpointReader(save_path)
473 (save_path, base.OBJECT_GRAPH_PROTO_KEY))
1025 sess=session, save_path=self._checkpoint.save_path)
[all …]
|
| D | python_state_test.py | 142 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)
|
| D | util_with_v1_optimizers_test.py | 206 save_path = root_trackable.save(file_prefix=prefix)
212 status = root_trackable.restore(save_path=save_path).assert_consumed()
228 status = on_create_root.restore(save_path=save_path)
364 status = root.restore(save_path=checkpoint_path)
398 status = root.restore(save_path=manager.latest_checkpoint)
431 status = root.restore(save_path=checkpoint_path)
589 save_path = obj.save(checkpoint_prefix)
590 obj.restore(save_path)
592 obj.restore(save_path)
631 save_path = second_root_trackable.save(checkpoint_prefix)
[all …]
|
| D | base_test.py | 67 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)
|
| D | data_structures_test.py | 110 save_path = os.path.join(self.get_temp_dir(), "ckpt")
111 model.save_weights(save_path)
113 model.load_weights(save_path)
571 save_path = os.path.join(self.get_temp_dir(), "ckpt")
572 model.save_weights(save_path)
574 model.load_weights(save_path)
635 save_path = os.path.join(self.get_temp_dir(), "ckpt")
637 model.save_weights(save_path)
646 save_path = os.path.join(self.get_temp_dir(), "ckpt")
647 model.save_weights(save_path)
[all …]
|
| /external/fonttools/Tests/cffLib/ |
| D | cffLib_test.py | 60 save_path = os.path.join(self.tempdir, 'TestOTF.otf')
61 font.save(save_path)
63 font2 = TTFont(save_path)
88 save_path = os.path.join(self.tempdir, 'TestOTF.otf')
89 font.save(save_path)
91 font2 = TTFont(save_path)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | values_test.py | 392 save_path, _ = self._save_return_saver(sess, var)
393 return save_path
413 save_path, saver = self._save_return_saver(sess, mirrored)
419 saver.restore(sess, save_path)
431 save_path = self._save(sess, mirrored)
435 return save_path
447 save_path = self._save(sess, var)
451 return save_path
453 def _restore_normal(self, save_path): argument
464 saver.restore(sess, save_path)
[all …]
|
| D | checkpointing_test.py | 133 save_path = checkpoint.save(prefix)
134 return save_path
136 save_path = checkpoint()
143 checkpoint.restore(save_path)
154 checkpoint.restore(save_path)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver_test.py | 87 save_path = os.path.join(self.get_temp_dir(), "basic_save_restore")
114 val = save.save(sess, save_path)
116 self.assertEqual(save_path, val)
133 save.restore(sess, save_path)
161 save2.restore(sess, save_path)
273 save_path = os.path.join(self.get_temp_dir(), "resource_cache")
282 save.save(sess, save_path)
285 save2.restore(sess, save_path)
397 save_path = os.path.join(self.get_temp_dir(), "int64")
406 val = save.save(sess, save_path)
[all …]
|
| D | saver_large_partitioned_variable_test.py | 39 save_path = os.path.join(self.get_temp_dir(), "large_variable")
56 val = save.save(sess, save_path)
57 self.assertEqual(save_path, val)
|
| D | saver.py | 1076 save_path, argument
1150 checkpoint_file = "%s-%d" % (save_path, global_step)
1153 checkpoint_file = "%s-%s" % (save_path, "{:08d}".format(global_step))
1155 checkpoint_file = save_path
1156 if os.path.basename(save_path) == latest_filename and not self._sharded:
1160 (latest_filename, save_path))
1166 save_path_parent = os.path.dirname(save_path)
1192 save_path))
1256 def restore(self, sess, save_path): argument
1276 if save_path is None:
[all …]
|
| D | quantize_training_test.py | 58 save_path = os.path.join(self.get_temp_dir(), 'quantized_save_restore')
84 saver.save(sess, save_path)
91 saver.restore(sess, save_path)
|
| D | checkpoint_management.py | 734 save_path = self._checkpoint.write(prefix)
738 if save_path in self._maybe_delete:
739 del self._maybe_delete[save_path]
740 self._maybe_delete[save_path] = timestamp
741 self._latest_checkpoint = save_path
751 return save_path
|
| D | supervisor_test.py | 436 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 …]
|
| D | saver_large_variable_test.py | 40 save_path = os.path.join(self.get_temp_dir(), "large_variable")
56 save.save(sess, save_path)
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | quantile_ops_test.py | 182 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
201 save.save(sess, save_path)
207 save.restore(sess, save_path)
214 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
229 save.save(sess, save_path)
238 save.restore(sess, save_path)
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | checkpoint_test.py | 303 save_path = checkpoint.save(checkpoint_prefix)
306 checkpoint.restore(save_path).run_restore_ops()
332 save_path = checkpoint.save(checkpoint_prefix)
336 checkpoint.restore(save_path).run_restore_ops()
351 save_path = checkpoint.save(checkpoint_prefix)
353 checkpoint.restore(save_path).run_restore_ops()
355 save_path = checkpoint.save(checkpoint_prefix)
356 checkpoint.restore(save_path).run_restore_ops()
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver_test.py | 75 save_path = save(constant_op.constant(prefix))
77 restore(constant_op.constant(save_path))
83 second_saver.restore(save_path)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | async_checkpoint.py | 67 save_path = os.path.join(checkpoint_dir, checkpoint_basename)
69 save_path)
78 self._save_path = save_path
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-saver.pbtxt | 35 argspec: "args=[\'self\', \'sess\', \'save_path\'], varargs=None, keywords=None, defaults=None"
39 …argspec: "args=[\'self\', \'sess\', \'save_path\', \'global_step\', \'latest_filename\', \'meta_gr…
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | iterator_ops.py | 306 save_path, argument
314 sess, save_path, global_step, latest_filename or self._latest_filename,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | partitioned_variables_test.py | 623 save_path = saver.save(sess=session, save_path=save_prefix)
630 saver = saver_lib.import_meta_graph(save_path + ".meta")
631 saver.restore(sess=session, save_path=save_path)
|
| D | lookup_ops_test.py | 1348 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
1378 val = save.save(sess, save_path)
1380 self.assertEqual(save_path, val)
1401 save.restore(sess, save_path)
1413 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
1443 val = save.save(sess, save_path)
1445 self.assertEqual(save_path, val)
1466 save.restore(sess, save_path)
1502 save_path = save_checkpoint.save(save_prefix)
1524 restore_checkpoint.restore(save_path).run_restore_ops()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
| D | keras_test.py | 286 save_path = checkpoint.save(prefix)
297 checkpoint.restore(save_path).assert_consumed().run_restore_ops()
561 save_path = os.path.join(self.get_temp_dir(), 'model')
562 model.save(save_path, save_format=save_format)
563 model = save.load_model(save_path)
579 model = save.load_model(save_path)
1067 save_path = os.path.join(self.get_temp_dir(), 'model')
1068 model.save(save_path, save_format='h5' if h5 else 'tf')
1079 save_path, custom_objects={'MultiplyLayer': mp_test_util.MultiplyLayer})
|