Searched refs:output_saved_model_dir (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| H A D | trt_convert_test.py | 179 output_saved_model_dir=None, argument
188 if output_saved_model_dir:
231 if output_saved_model_dir is not None:
232 converter.save(output_saved_model_dir=output_saved_model_dir)
255 output_saved_model_dir=None, argument
260 output_saved_model_dir=output_saved_model_dir,
266 if output_saved_model_dir:
268 output_saved_model_dir, tag_constants.SERVING).graph_def
339 output_saved_model_dir=self.mkdtemp(),
436 output_saved_model_dir = self.mkdtemp()
[all …]
|
| H A D | trt_convert.py | 761 def save(self, output_saved_model_dir): argument
828 saved_model_builder = builder.SavedModelBuilder(output_saved_model_dir)
1497 output_saved_model_dir, argument
1587 self._saved_model, output_saved_model_dir, signatures, options=options)
1707 output_saved_model_dir=None): argument
1775 if output_saved_model_dir:
1776 trt_converter.save(output_saved_model_dir)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| H A D | model_handler.py | 435 output_saved_model_dir: Optional[str] = None,
440 output_saved_model_dir = output_saved_model_dir or tempfile.mkdtemp()
441 logging.info("Saving TensorRT model to %s!", output_saved_model_dir)
442 self._converter.save(output_saved_model_dir)
444 saved_model_dir=output_saved_model_dir)
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.experimental.tensorrt.-converter.pbtxt | 19 …argspec: "args=[\'self\', \'output_saved_model_dir\', \'save_gpu_specific_engines\', \'options\'],…
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tools/ |
| H A D | saved_model_cli.py | 828 converter.save(output_saved_model_dir=args.output_dir)
840 output_saved_model_dir=args.output_dir)
|