| /aosp_15_r20/external/federated-compute/fcp/tensorflow/ |
| H A D | append_slices_test.py | 30 checkpoint_path = self.new_tempfile_path()
34 filename=checkpoint_path,
38 append_slices.merge_appended_slices(checkpoint_path)
40 prefix=checkpoint_path,
47 checkpoint_path = self.new_tempfile_path()
52 filename=checkpoint_path,
56 append_slices.merge_appended_slices(checkpoint_path)
58 prefix=checkpoint_path,
66 checkpoint_path = self.new_tempfile_path()
69 filename=checkpoint_path,
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | evaluation_test.py | 118 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
121 checkpoint_path=checkpoint_path,
150 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
153 checkpoint_path=checkpoint_path,
173 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
189 checkpoint_path=checkpoint_path,
201 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
219 checkpoint_path=checkpoint_path,
230 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
242 checkpoint_path=checkpoint_path, final_ops={'value': final_ops})
|
| H A D | checkpoint_utils.py | 143 checkpoint_path = checkpoint_management.latest_checkpoint(checkpoint_dir)
144 if checkpoint_path is None or checkpoint_path == last_checkpoint:
149 logging.info("Found new checkpoint at %s", checkpoint_path)
150 return checkpoint_path
196 checkpoint_path = None
199 checkpoint_dir, checkpoint_path, timeout=timeout)
212 checkpoint_path = new_checkpoint_path
213 yield checkpoint_path
|
| H A D | saver.py | 945 def _build_eager(self, checkpoint_path, build_save, build_restore): argument
947 checkpoint_path, build_save=build_save, build_restore=build_restore)
949 def _build(self, checkpoint_path, build_save, build_restore): argument
979 filename=checkpoint_path,
1171 for checkpoint_path in checkpoint_paths:
1173 mtime = checkpoint_management.get_checkpoint_mtimes([checkpoint_path])
1179 checkpoints_with_mtimes.append((checkpoint_path, mtime[0]))
1442 checkpoint_path=save_path,
1738 def object_graph_key_mapping(checkpoint_path): argument
1747 reader = py_checkpoint_reader.NewCheckpointReader(checkpoint_path)
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/ |
| H A D | saveable_compat_test.py | 42 checkpoint_path = os.path.join(checkpoint_directory, "ckpt")
43 ckpt.write(checkpoint_path)
47 metadata = checkpoint.object_metadata(checkpoint_path)
63 reader = checkpoint_utils.load_checkpoint(checkpoint_path)
81 checkpoint_path = os.path.join(checkpoint_directory, "ckpt")
82 ckpt.write(checkpoint_path)
88 new_ckpt.read(checkpoint_path).assert_consumed()
146 checkpoint_path = os.path.join(self.get_temp_dir(), "ckpt")
147 checkpoint.Checkpoint(new).write(checkpoint_path)
150 checkpoint.Checkpoint(dep).read(checkpoint_path).assert_consumed()
[all …]
|
| H A D | benchmarks_test.py | 93 checkpoint_path = _save_checkpoint()
98 checkpoint.restore(checkpoint_path)
103 checkpoint_path = _save_checkpoint()
107 checkpoint.restore(checkpoint_path)
113 checkpoint_path = _save_checkpoint()
115 checkpoint_path).get_variable_to_dtype_map().items())
118 gen_io_ops.restore_v2(checkpoint_path, all_names, [""] * len(all_names),
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tools/ |
| H A D | freeze_graph_test.py | 67 checkpoint_path = saver.save(
89 checkpoint_path,
183 checkpoint_path = saver.save(
195 input_meta_graph = checkpoint_path + ".meta"
198 "", input_saver_def_path, input_binary, checkpoint_path,
238 checkpoint_path = None
243 input_binary, checkpoint_path, output_node_names,
306 checkpoint_path = saver.save(
329 input_checkpoint=checkpoint_path,
|
| H A D | tools.bzl | 18 checkpoint_path = None,
77 files pointed to by `directory` and `checkpoint_path`.
80 checkpoint_path: The bazel directory containing `variables.index`. If
115 if checkpoint_path:
117 "--checkpoint_path \"$$(dirname $(location {}/variables.index))\" "
118 .format(checkpoint_path)
120 checkpoint_srcs = ["{}/variables.index".format(checkpoint_path)]
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.estimator.-estimator.pbtxt | 31 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
35 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
39 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
43 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
59 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-baseline-regressor.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-baseline-classifier.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-baseline-estimator.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-linear-estimator.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-linear-regressor.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-d-n-n-linear-combined-regressor.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-d-n-n-regressor.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-d-n-n-linear-combined-estimator.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-linear-classifier.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.tpu.-t-p-u-estimator.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-d-n-n-classifier.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-d-n-n-estimator.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.-d-n-n-linear-combined-classifier.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.experimental.-k-means.pbtxt | 68 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
72 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
76 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
80 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
96 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.estimator.experimental.-r-n-n-estimator.pbtxt | 32 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
36 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
40 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
44 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
60 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|
| H A D | tensorflow.estimator.experimental.-r-n-n-classifier.pbtxt | 33 …argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], vara…
37 …_base\', \'input_receiver_fn_map\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=N…
41 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'experime…
45 …e\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_de…
61 …argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_…
|