| /external/tensorflow/tensorflow/python/saved_model/ |
| D | save_test.py | 79 save_dir, argument
85 model = loader.load(session, [tag_constants.SERVING], save_dir)
97 save_dir = os.path.join(self.get_temp_dir(), "saved_model")
98 save.save(root, save_dir, root.f)
99 self.assertEqual({"output_0": 2.}, _import_and_infer(save_dir, {"x": 1.}))
115 save_dir = os.path.join(self.get_temp_dir(), "saved_model")
116 save.save(root, save_dir, root.f)
117 self.assertEqual({"output_0": 4.}, _import_and_infer(save_dir, {"x": 1.}))
123 save_dir = os.path.join(self.get_temp_dir(), "saved_model")
125 root, save_dir, {
[all …]
|
| D | load_test.py | 218 save_dir = os.path.join(self.get_temp_dir(), "save_dir")
219 save.save(root, save_dir)
224 file_io.rename(save_dir, load_dir)
979 save_dir = os.path.join(self.get_temp_dir(), "saved_model")
980 save.save(root, save_dir)
2066 save_dir = os.path.join(self.get_temp_dir(), "saved_model")
2067 save.save(obj, save_dir, signatures=total.get_concrete_function())
2068 imported = load.load(save_dir)
2078 imported = load.load(save_dir)
2115 save_dir = os.path.join(self.get_temp_dir(), "saved_model")
[all …]
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/saved_model_benchmarks/ |
| D | saved_model_benchmark_util.py | 39 save_dir = tempfile.mkdtemp(dir=tmp_dir)
45 model.save(save_dir, save_format='tf')
46 tf.keras.models.load_model(save_dir)
50 model.save(save_dir, save_format='tf')
54 tf.keras.models.load_model(save_dir)
68 gfile.DeleteRecursively(save_dir)
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | make_aot_compile_models.py | 50 save_dir = os.path.join(out_dir, 'x_matmul_y_large')
51 save.save(root, save_dir, root.f)
54 with open(os.path.join(save_dir, 'variables', 'variables.index'), 'w'):
67 save_dir = os.path.join(out_dir, 'x_matmul_y_small')
68 save.save(root, save_dir, root.f)
71 with open(os.path.join(save_dir, 'variables', 'variables.index'), 'w'):
|
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_management.py | 50 def _GetCheckpointFilename(save_dir, latest_filename): argument
63 return os.path.join(save_dir, latest_filename)
67 def generate_checkpoint_state_proto(save_dir, argument
115 if not os.path.isabs(save_dir):
117 model_checkpoint_path = os.path.relpath(model_checkpoint_path, save_dir)
120 all_model_checkpoint_paths[i] = os.path.relpath(p, save_dir)
136 def update_checkpoint_state(save_dir, argument
168 save_dir=save_dir,
177 def update_checkpoint_state_internal(save_dir, argument
213 coord_checkpoint_filename = _GetCheckpointFilename(save_dir, latest_filename)
[all …]
|
| D | checkpoint_management_test.py | 158 save_dir = self._get_test_dir("abs_paths")
159 abs_path = os.path.join(save_dir, "model-0")
161 save_dir, abs_path)
179 save_dir = self._get_test_dir("all_models_test")
180 abs_path = os.path.join(save_dir, "model-0")
183 save_dir, abs_path, all_model_checkpoint_paths=paths)
191 save_dir = self._get_test_dir("update_checkpoint_state")
192 os.chdir(save_dir)
196 abs_path = os.path.join(save_dir, "model-0")
207 save_dir = self._get_test_dir("update_checkpoint_state")
[all …]
|
| D | saver_test.py | 822 save_dir = self.get_temp_dir()
837 save.save(sess, save_dir)
845 save.restore(sess, save_dir)
1249 all_model_checkpoint_paths, save_dir): argument
1250 checkpoint_state = checkpoint_management.get_checkpoint_state(save_dir)
1258 save_dir = self._get_test_dir("max_to_keep_eager")
1266 s1 = save.save(None, os.path.join(save_dir, "s1"))
1272 save_dir=save_dir)
1274 s2 = save.save(None, os.path.join(save_dir, "s2"))
1281 save_dir=save_dir)
[all …]
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | saved_model_test.py | 62 def _import_and_infer(save_dir, inputs, signature_key="serving_default"): argument
66 model = tf.compat.v1.saved_model.load(session, ["serve"], save_dir)
97 save_dir = os.path.join(self.get_temp_dir(), "saved_model")
98 tf.saved_model.save(model, save_dir, model.call)
103 _import_and_infer(save_dir, {"x": [[3., 4.]], "y": [2.]}))
110 save_dir = os.path.join(self.get_temp_dir(), "saved_model")
111 tf.saved_model.save(model, save_dir)
113 _import_and_infer(save_dir,
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils_test.py | 251 def _import_and_infer(save_dir, inputs): argument
255 model = loader.load(session, [tag_constants.SERVING], save_dir)
327 save_dir = os.path.join(self.get_temp_dir(), 'saved_model')
328 save_lib.save(model, save_dir)
338 _import_and_infer(save_dir,
344 load_lib.load(save_dir)(inputs))
354 save_dir = os.path.join(self.get_temp_dir(), 'saved_model')
355 save_lib.save(model, save_dir)
365 _import_and_infer(save_dir,
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_v2_test.py | 913 save_dir = os.path.join(self.get_temp_dir(), 'saved_model')
914 save(root, save_dir, to_save)
917 converter = lite.TFLiteConverterV2.from_saved_model(save_dir)
932 save_dir = os.path.join(self.get_temp_dir(), 'saved_model')
933 save(root, save_dir, to_save)
936 converter = lite.TFLiteConverterV2.from_saved_model(save_dir)
951 save_dir = os.path.join(self.get_temp_dir(), 'saved_model')
952 save(root, save_dir, to_save)
957 save_dir, signature_keys=['INVALID'])
962 save_dir, signature_keys=[])
[all …]
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | saved_model_test.py | 72 save_dir = os.path.join(self.get_temp_dir(), "saved_model")
73 save.save(self._model, save_dir, self._model.call)
|
| D | convert_to_constants_test.py | 93 save_dir = os.path.join(self.get_temp_dir(), "frozen_saved_model")
96 save(root, save_dir, {"mykey": converted_concrete_func})
99 loaded_obj = load(save_dir)
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | quantile_ops_test.py | 181 save_dir = os.path.join(self.get_temp_dir(), "save_restore")
182 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
219 save_dir = os.path.join(self.get_temp_dir(), "save_restore")
220 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
|
| /external/autotest/server/cros/network/ |
| D | packet_capturer.py | 423 remote_pcap, remote_pcap_log, save_dir = self._ongoing_captures[pid]
427 pcap_filename = os.path.join(local_save_dir or save_dir,
429 pcap_log_filename = os.path.join(local_save_dir or save_dir,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable_test.py | 296 save_dir = os.path.join(self.get_temp_dir(), 'saved_model')
297 save.save(root, save_dir, root.serve)
299 _load_and_run(save_dir, {'x': [0, 1]})['output_0'])
310 save_dir = os.path.join(self.get_temp_dir(), 'saved_model')
311 save.save(root, save_dir)
315 load.load(save_dir)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | convert_to_constants_test.py | 197 save_dir = os.path.join(self.get_temp_dir(), "frozen_saved_model")
200 save(root, save_dir, {"mykey": converted_concrete_func})
203 loaded_obj = load(save_dir)
216 save_dir = os.path.join(self.get_temp_dir(), "saved_model")
217 save(root, save_dir, to_save)
218 saved_model = load(save_dir)
273 save_dir = os.path.join(self.get_temp_dir(), "saved_model")
274 save(root, save_dir, to_save)
275 saved_model = load(save_dir)
|
| /external/autotest/server/site_tests/network_WiFi_ChaosLongConnect/ |
| D | network_WiFi_ChaosLongConnect.py | 74 capturer.stop_capture(save_dir=self.outputdir,
|
| /external/autotest/server/site_tests/network_WiFi_ChaosConnectDisconnect/ |
| D | network_WiFi_ChaosConnectDisconnect.py | 65 capturer.stop_capture(save_dir=self.outputdir,
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | saved_model_test.py | 1026 save_dir, argument
1031 tf_save.save(metric, save_dir)
1032 loaded = keras_load.load(save_dir)
1097 save_dir = self._save_model_dir('first_save')
1106 self._test_metric_save_and_load(metric, save_dir, num_tensor_args)
1110 save_dir,
1133 save_dir = self._save_model_dir('first_save')
1137 save_dir,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | model.h | 646 const char* save_dir = std::getenv("TF_DATA_AUTOTUNE_DEBUG_DIR"); in Model() local
647 if (save_dir) { in Model()
648 save_dir_ = string(save_dir); in Model()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | lookup_ops_test.py | 1577 save_dir = os.path.join(self.get_temp_dir(), "save_restore")
1578 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
1642 save_dir = os.path.join(self.get_temp_dir(), "save_restore")
1643 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
1707 save_dir = os.path.join(self.get_temp_dir(), "save_restore")
1708 save_prefix = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
1765 save_dir = os.path.join(self.get_temp_dir(), "save_restore")
1766 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
1806 save_dir = os.path.join(self.get_temp_dir(), "vector_save_restore")
1807 save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
[all …]
|
| /external/autotest/server/ |
| D | site_linux_system.py | 394 def stop_capture(self, save_dir=None, save_filename=None): argument
404 local_save_dir=save_dir, local_pcap_filename=save_filename)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 317 …argspec: "args=[\'save_dir\', \'model_checkpoint_path\', \'all_model_checkpoint_paths\', \'all_mod…
469 …argspec: "args=[\'save_dir\', \'model_checkpoint_path\', \'all_model_checkpoint_paths\', \'latest_…
|
| /external/autotest/server/cros/clique_lib/ |
| D | clique_dut_control.py | 501 capturer.stop_capture(save_dir=self.outputdir,
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util.py | 1666 save_dir=os.path.dirname(file_prefix),
2103 save_dir=os.path.dirname(file_prefix),
|