Searched refs:minimum_segment_size (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 158 minimum_segment_size=3, argument
164 minimum_segment_size, maximum_cached_engines,
289 "minimum_segment_size"].i = conversion_params.minimum_segment_size
380 minimum_segment_size=3, argument
474 minimum_segment_size=minimum_segment_size,
1242 minimum_segment_size=3, argument
1311 minimum_segment_size=minimum_segment_size,
|
| D | trt_convert_windows.py | 85 minimum_segment_size=3, argument
|
| D | trt_convert_test.py | 180 minimum_segment_size=3, argument
208 minimum_segment_size=minimum_segment_size,
767 output_graph_def = self._ConvertGraphV1(minimum_segment_size=7)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.h | 41 int minimum_segment_size = 3; member
|
| D | convert_graph_test.cc | 169 params.minimum_segment_size = 1; in RunConvertAfterShape()
|
| D | trt_optimization_pass.cc | 239 cp.minimum_segment_size = minimum_segment_size_; in Optimize()
|
| D | convert_graph.cc | 734 segment_options.minimum_segment_size = params.minimum_segment_size; in ConvertAfterShapes()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.experimental.tensorrt.-conversion-params.pbtxt | 19 name: "minimum_segment_size"
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.h | 41 int minimum_segment_size = 2; member
|
| D | segment.cc | 1045 num_effective_nodes < options.minimum_segment_size) { in SegmentGraph()
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_cli.py | 784 minimum_segment_size=args.minimum_segment_size)
802 minimum_segment_size=args.minimum_segment_size,
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | model_handler.py | 461 minimum_segment_size=trt_convert_params.minimum_segment_size,
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 282 minimum_segment_size=2,
468 minimum_segment_size=conversion_params.minimum_segment_size,
|
| D | quantization_mnist_test.py | 155 minimum_segment_size=2,
|