Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert_test.py158 def _WriteInputSavedModelForV1(self, input_saved_model_dir, device): argument
168 saved_model_builder = builder.SavedModelBuilder(input_saved_model_dir)
185 input_saved_model_dir = None
187 input_saved_model_dir = self.mkdtemp()
188 self._WriteInputSavedModelForV1(input_saved_model_dir, device)
199 input_saved_model_dir=input_saved_model_dir,
202 if input_saved_model_dir else self._GetGraphDefForV1(device),
203 nodes_denylist=None if input_saved_model_dir else ["output"],
379 input_saved_model_dir, argument
385 input_saved_model_dir=input_saved_model_dir,
[all …]
Dtrt_convert.py372 input_saved_model_dir=None, argument
427 if input_graph_def and input_saved_model_dir:
430 if not input_graph_def and not input_saved_model_dir:
438 self._input_saved_model_dir = input_saved_model_dir
927 input_saved_model_dir=None, argument
952 self._input_saved_model_dir = input_saved_model_dir
1245 input_saved_model_dir=None, argument
1303 input_saved_model_dir=input_saved_model_dir,
Dtrt_convert_windows.py101 input_saved_model_dir=None, argument
/external/tensorflow/tensorflow/python/tools/
Dfreeze_graph.py89 input_saved_model_dir=None, argument
126 if (not input_saved_model_dir and
158 elif input_saved_model_dir:
161 loader.load(sess, saved_model_tags, input_saved_model_dir)
299 input_saved_model_dir=None, argument
334 if input_saved_model_dir:
336 input_saved_model_dir, saved_model_tags).graph_def
359 input_saved_model_dir,
377 flags.input_meta_graph, flags.input_saved_model_dir,
Dfreeze_graph_test.py234 input_saved_model_dir = saved_model_dir
250 input_meta_graph, input_saved_model_dir,
Dsaved_model_cli.py786 input_saved_model_dir=args.dir,
804 input_saved_model_dir=args.dir,
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/testdata/
Dgen_tftrt_model.py85 input_saved_model_dir=tf_saved_model_dir)
129 input_saved_model_dir=tf_saved_model_dir, is_dynamic_op=True)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.experimental.tensorrt.-converter.pbtxt7 …argspec: "args=[\'self\', \'input_saved_model_dir\', \'input_saved_model_tags\', \'input_saved_mod…
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Dmodel_handler.py454 input_saved_model_dir=self.model_config.saved_model_dir,
508 input_saved_model_dir=self.model_config.saved_model_dir,
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py455 input_saved_model_dir=saved_model_dir,
464 input_saved_model_dir=saved_model_dir,