Searched refs:input_saved_model_dir (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert_test.py | 154 def _WriteInputSavedModel(self, input_saved_model_dir): argument
161 saved_model_builder = builder.SavedModelBuilder(input_saved_model_dir)
170 input_saved_model_dir=None, argument
180 input_saved_model_dir=input_saved_model_dir,
182 input_graph_def=None if input_saved_model_dir else self._GetGraphDef(),
221 input_saved_model_dir=None, argument
227 input_saved_model_dir=input_saved_model_dir,
286 input_saved_model_dir = os.path.join(tmp_dir, "in_dir1")
287 self._WriteInputSavedModel(input_saved_model_dir)
297 input_saved_model_dir=input_saved_model_dir,
[all …]
|
| D | trt_convert.py | 104 input_saved_model_dir=None, argument
131 if input_graph_def or not input_saved_model_dir:
136 if input_graph_def and input_saved_model_dir:
139 if not input_graph_def and not input_saved_model_dir:
146 self._input_saved_model_dir = input_saved_model_dir
558 input_saved_model_dir=None, argument
626 input_saved_model_dir=input_saved_model_dir,
791 input_saved_model_dir=None, argument
859 input_saved_model_dir=input_saved_model_dir,
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | freeze_graph.py | 89 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)
301 input_saved_model_dir=None, argument
336 if input_saved_model_dir:
338 input_saved_model_dir, saved_model_tags).graph_def
361 input_saved_model_dir,
380 flags.input_meta_graph, flags.input_saved_model_dir,
|
| D | freeze_graph_test.py | 228 input_saved_model_dir = saved_model_dir
244 input_meta_graph, input_saved_model_dir,
|
| D | saved_model_cli.py | 678 input_saved_model_dir=args.dir,
|
| /external/tensorflow/tensorflow/contrib/tensorrt/python/ |
| D | trt_convert.py | 34 input_saved_model_dir=None, argument
48 input_saved_model_dir=input_saved_model_dir,
|