Searched defs:precision_mode (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| H A D | trt_convert.py | 172 precision_mode=TrtPrecisionMode.FP32, argument
437 precision_mode=TrtPrecisionMode.FP32, argument
1150 precision_mode=TrtPrecisionMode.FP32, argument
1700 precision_mode=TrtPrecisionMode.FP32, argument
|
| H A D | trt_convert_test.py | 380 precision_mode=trt_convert.TrtPrecisionMode.FP32, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.h | 133 TrtPrecisionMode precision_mode; member
308 TrtPrecisionMode precision_mode() const { return precision_mode_; } in precision_mode() function
|
| H A D | trt_optimization_pass.h | 47 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32; member
|
| H A D | trt_optimization_pass.cc | 94 std::string precision_mode; in UpdateFunctionSpecificConversionParams() local
|
| H A D | op_converter.h | 62 const TrtPrecisionMode precision_mode; member
|
| H A D | convert_nodes.cc | 841 TrtPrecisionMode precision_mode, bool use_calibration, in OpConverterParams()
870 TrtPrecisionMode precision_mode, bool use_calibration, in TrtNodeValidator()
1020 TrtPrecisionMode precision_mode, bool use_calibration, in Create()
1031 Converter::Converter(TrtPrecisionMode precision_mode, bool use_calibration, in Converter()
5830 const GraphDef& gdef, OpKernelContext* ctx, TrtPrecisionMode precision_mode, in ConvertGraphDefToEngine()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| H A D | trt_convert_api.h | 48 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32; member
|