Home
last modified time | relevance | path

Searched refs:checkFloatTensor (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DAffineQuantizer.cpp30 void checkFloatTensor(const std::string& fn_name, const Tensor& t) { in checkFloatTensor() function
117 checkFloatTensor(fn_name, rtensor); in quantize_tensor_per_tensor_affine()
147 checkFloatTensor(fn_name, rtensor); in quantize_tensor_per_channel_affine()
183 checkFloatTensor(fn_name, rtensor); in quantize_tensor_per_channel_float_qparams()
211 checkFloatTensor(fn_name, rtensor); in dequantize_tensor_per_tensor_affine()
238 checkFloatTensor(fn_name, rtensor); in dequantize_tensor_per_channel_affine()
272 checkFloatTensor(fn_name, rtensor); in dequantize_tensor_per_channel_float_qparams()