Home
last modified time | relevance | path

Searched defs:rtensor (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/quantized/
H A DQuantizer.cpp163 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 DAffineQuantizer.cpp93 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 DAffineQuantizer.cu41 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 DQuantizedOpKernels.cpp3317 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 Dutils.h130 const executorch::aten::Tensor& rtensor, in QuantizePerTensor()
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp123 const at::Tensor& rtensor, in quantize_tensor_per_tensor_affine_privateuse1()
/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Ddim.cpp2099 at::Tensor rtensor; in invoke_getitem() local