Searched refs:output_saved_model_dir (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert_test.py | 177 output_saved_model_dir=None, argument
186 if output_saved_model_dir:
229 if output_saved_model_dir is not None:
230 converter.save(output_saved_model_dir=output_saved_model_dir)
253 output_saved_model_dir=None, argument
258 output_saved_model_dir=output_saved_model_dir,
264 if output_saved_model_dir:
266 output_saved_model_dir, tag_constants.SERVING).graph_def
339 output_saved_model_dir=self.mkdtemp(),
440 output_saved_model_dir = self.mkdtemp()
[all …]
|
| D | trt_convert.py | 703 def save(self, output_saved_model_dir): argument
770 saved_model_builder = builder.SavedModelBuilder(output_saved_model_dir)
1153 def save(self, output_saved_model_dir): argument
1232 save.save(self._saved_model, output_saved_model_dir, signatures)
1248 output_saved_model_dir=None): argument
1316 if output_saved_model_dir:
1317 trt_converter.save(output_saved_model_dir)
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| 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)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.experimental.tensorrt.-converter.pbtxt | 19 … argspec: "args=[\'self\', \'output_saved_model_dir\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_cli.py | 794 converter.save(output_saved_model_dir=args.output_dir)
806 output_saved_model_dir=args.output_dir)
|