Searched defs:precision_mode (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert_windows.py | 84 precision_mode=TrtPrecisionMode.FP32, argument
|
| D | trt_convert.py | 157 precision_mode=TrtPrecisionMode.FP32, argument
379 precision_mode=TrtPrecisionMode.FP32, argument
1241 precision_mode=TrtPrecisionMode.FP32, argument
|
| D | trt_convert_test.py | 382 precision_mode=trt_convert.TrtPrecisionMode.FP32, argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 115 TrtPrecisionMode precision_mode; member
379 const TrtPrecisionMode precision_mode; member
494 TrtPrecisionMode precision_mode() const { return precision_mode_; } in precision_mode() function
|
| D | convert_graph.h | 40 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32; member
|
| D | convert_nodes.cc | 1104 TrtPrecisionMode precision_mode, bool use_calibration, in OpConverterParams()
1138 TrtPrecisionMode precision_mode, bool use_calibration, in TrtNodeValidator()
1243 TrtPrecisionMode precision_mode, bool use_calibration, in Create()
1253 Converter::Converter(TrtPrecisionMode precision_mode, bool use_calibration, in Converter()
6320 const GraphDef& gdef, TrtPrecisionMode precision_mode, int max_batch_size, in ConvertGraphDefToEngine()
|