Searched refs:output_saved_model_dir (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert_test.py | 171 output_saved_model_dir=None, argument
216 if output_saved_model_dir is not None:
217 converter.save(output_saved_model_dir=output_saved_model_dir)
222 output_saved_model_dir=None, argument
228 output_saved_model_dir=output_saved_model_dir,
234 if output_saved_model_dir:
236 root = load.load(output_saved_model_dir)
242 output_saved_model_dir, tag_constants.SERVING).graph_def
294 output_saved_model_dir = os.path.join(
298 output_saved_model_dir=output_saved_model_dir,
[all …]
|
| D | trt_convert.py | 398 def save(self, output_saved_model_dir): argument
420 save.save(self._saved_model, output_saved_model_dir, signatures)
427 saved_model_builder = builder.SavedModelBuilder(output_saved_model_dir)
774 def save(self, output_saved_model_dir): argument
778 super(TrtGraphConverter, self).save(output_saved_model_dir)
794 output_saved_model_dir=None, argument
874 if output_saved_model_dir:
875 trt_converter.save(output_saved_model_dir)
|
| /external/tensorflow/tensorflow/contrib/tensorrt/python/ |
| D | trt_convert.py | 36 output_saved_model_dir=None, argument
50 output_saved_model_dir=output_saved_model_dir,
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_cli.py | 680 output_saved_model_dir=args.output_dir)
|