Searched defs:rtensor (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/quantized/ |
H A D | Quantizer.cpp | 163 Tensor PerTensorAffineQuantizer::quantize(const Tensor& rtensor) { in quantize() 183 Tensor& rtensor, in per_tensor_affine_dequantize_impl() 194 Tensor& rtensor, const Tensor& qtensor) { in dequantize_out() 208 Tensor rtensor = at::empty( in dequantize() local 217 Tensor PerChannelAffineQuantizer::quantize(const Tensor& rtensor) { in quantize() 233 Tensor& rtensor, in per_channel_affine_dequantize_impl() 245 Tensor rtensor = at::empty( in dequantize() local 255 Tensor& rtensor, const Tensor& qtensor) { in dequantize_out() 268 Tensor PerChannelAffineFloatQParamsQuantizer::quantize(const Tensor& rtensor) { in quantize() 283 Tensor& rtensor, in per_channel_affine_float_q_params_dequantize_impl() [all …]
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/ |
H A D | AffineQuantizer.cpp | 93 const Tensor& rtensor, in checkPerChannelParamsSize() 110 const Tensor& rtensor, in quantize_tensor_per_tensor_affine() 139 const Tensor& rtensor, in quantize_tensor_per_channel_affine() 174 const Tensor& rtensor, in quantize_tensor_per_channel_float_qparams() 207 Tensor& rtensor, in dequantize_tensor_per_tensor_affine() 232 Tensor& rtensor, in dequantize_tensor_per_channel_affine() 266 Tensor& rtensor, in dequantize_tensor_per_channel_float_qparams()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/ |
H A D | AffineQuantizer.cu | 41 const Tensor& rtensor, in quantize_tensor_per_tensor_affine_cuda() 71 Tensor& rtensor, in dequantize_tensor_per_tensor_affine_cuda() 88 const Tensor& rtensor, in quantize_tensor_per_channel_affine_cuda() 132 Tensor& rtensor, in dequantize_tensor_per_channel_affine_cuda() 167 const Tensor& rtensor, in quantize_tensor_per_channel_float_qparams_cuda() 216 Tensor& rtensor, in dequantize_tensor_per_channel_float_qparams_cuda()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
H A D | QuantizedOpKernels.cpp | 3317 const Tensor& rtensor, 3351 Tensor& rtensor, 3651 const Tensor& rtensor, 3686 Tensor& rtensor, 3723 const Tensor& rtensor, 3782 const Tensor& rtensor, 3968 const Tensor& rtensor, 3988 Tensor& rtensor, 4052 Tensor& rtensor, 4064 const Tensor& rtensor, [all …]
|
/aosp_15_r20/external/executorch/backends/xnnpack/runtime/utils/ |
H A D | utils.h | 130 const executorch::aten::Tensor& rtensor, in QuantizePerTensor()
|
/aosp_15_r20/external/pytorch/test/cpp_extensions/ |
H A D | open_registration_extension.cpp | 123 const at::Tensor& rtensor, in quantize_tensor_per_tensor_affine_privateuse1()
|
/aosp_15_r20/external/pytorch/functorch/csrc/dim/ |
H A D | dim.cpp | 2099 at::Tensor rtensor; in invoke_getitem() local
|