Searched refs:minimum_segment_size (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensorrt/python/ |
| D | trt_convert.py | 30 minimum_segment_size=3, argument
44 minimum_segment_size=minimum_segment_size,
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 467 minimum_segment_size=3, argument
543 optimizer.parameter_map["minimum_segment_size"].i = minimum_segment_size
567 minimum_segment_size=3, argument
704 self._minimum_segment_size = minimum_segment_size
715 minimum_segment_size=self._minimum_segment_size,
787 minimum_segment_size=3, argument
868 minimum_segment_size=minimum_segment_size,
|
| D | trt_convert_test.py | 69 minimum_segment_size=10,
174 minimum_segment_size=3, argument
189 minimum_segment_size=minimum_segment_size,
359 output_graph_def = self._ConvertGraph(minimum_segment_size=5)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.h | 65 int minimum_segment_size = 3; member
88 int minimum_segment_size = 3, bool is_dyn_op = false,
|
| D | convert_graph.cc | 105 int minimum_segment_size, bool is_dyn_op, int max_cached_engines, in ConvertGraphDefToTensorRT() argument
159 parameters["minimum_segment_size"].set_i(minimum_segment_size); in ConvertGraphDefToTensorRT()
815 segment_options.minimum_segment_size = params.minimum_segment_size; in ConvertAfterShapes()
|
| D | trt_optimization_pass.cc | 261 cp.minimum_segment_size = minimum_segment_size_; in Optimize()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.h | 41 int minimum_segment_size = 2; member
|
| D | segment.cc | 681 if (static_cast<int>(segment_nodes.size()) < options.minimum_segment_size) { in SegmentGraph()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 181 minimum_segment_size=2,
226 conversion_params.minimum_segment_size,
308 minimum_segment_size=conversion_params.minimum_segment_size,
|
| D | quantization_mnist_test.py | 150 minimum_segment_size=2,
|
| /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
| D | plugin_test.py | 89 minimum_segment_size=2)
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_cli.py | 676 minimum_segment_size=args.minimum_segment_size,
|