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.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 …]
Dtrt_convert.py417 input_saved_model_dir=None, argument
472 if input_graph_def and input_saved_model_dir:
475 if not input_graph_def and not input_saved_model_dir:
483 self._input_saved_model_dir = input_saved_model_dir
981 input_saved_model_dir=None, argument
1013 self._input_saved_model_dir = input_saved_model_dir
1332 input_saved_model_dir=None, argument
1390 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.py793 input_saved_model_dir=args.dir,
811 input_saved_model_dir=args.dir,
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/testdata/
Dgen_tftrt_model.py88 input_saved_model_dir=tf_saved_model_dir)
132 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.py458 input_saved_model_dir=saved_model_dir,
467 input_saved_model_dir=saved_model_dir,