Home
last modified time | relevance | path

Searched refs:input_saved_model_dir (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A Dtrt_convert_test.py160 def _WriteInputSavedModelForV1(self, input_saved_model_dir, device): argument
170 saved_model_builder = builder.SavedModelBuilder(input_saved_model_dir)
187 input_saved_model_dir = None
189 input_saved_model_dir = self.mkdtemp()
190 self._WriteInputSavedModelForV1(input_saved_model_dir, device)
201 input_saved_model_dir=input_saved_model_dir,
204 if input_saved_model_dir else self._GetGraphDefForV1(device),
205 nodes_denylist=None if input_saved_model_dir else ["output"],
377 input_saved_model_dir, argument
384 input_saved_model_dir=input_saved_model_dir,
[all …]
H A Dtrt_convert.py430 input_saved_model_dir=None, argument
485 if input_graph_def and input_saved_model_dir:
488 if not input_graph_def and not input_saved_model_dir:
496 self._input_saved_model_dir = input_saved_model_dir
1144 input_saved_model_dir=None, argument
1213 self._input_saved_model_dir = input_saved_model_dir
1704 input_saved_model_dir=None, argument
1762 input_saved_model_dir=input_saved_model_dir,
/aosp_15_r20/external/tensorflow/tensorflow/python/tools/
H A Dfreeze_graph.py85 input_saved_model_dir=None, argument
122 if (not input_saved_model_dir and
154 elif input_saved_model_dir:
157 loader.load(sess, saved_model_tags, input_saved_model_dir)
295 input_saved_model_dir=None, argument
330 if input_saved_model_dir:
332 input_saved_model_dir, saved_model_tags).graph_def
355 input_saved_model_dir,
373 flags.input_meta_graph, flags.input_saved_model_dir,
H A Dfreeze_graph_test.py230 input_saved_model_dir = saved_model_dir
246 input_meta_graph, input_saved_model_dir,
H A Dsaved_model_cli.py821 input_saved_model_dir=args.dir,
838 input_saved_model_dir=args.dir,
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/testdata/
H A Dgen_tftrt_model.py84 input_saved_model_dir=tf_saved_model_dir)
133 input_saved_model_dir=tf_saved_model_dir, is_dynamic_op=True)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.experimental.tensorrt.-converter.pbtxt7 …argspec: "args=[\'self\', \'input_saved_model_dir\', \'input_saved_model_tags\', \'input_saved_mod…
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
H A Dmodel_handler.py454 input_saved_model_dir=self.model_config.saved_model_dir,
508 input_saved_model_dir=self.model_config.saved_model_dir,
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
H A Dtf_trt_integration_test_base.py469 input_saved_model_dir=saved_model_dir,
478 input_saved_model_dir=saved_model_dir,
H A Dquantization_mnist_test.py347 input_saved_model_dir=saved_model_dir,