Home
last modified time | relevance | path

Searched defs:qtensor (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/quantized/
H A DQuantizer.cpp169 Tensor qtensor = new_qtensor( in quantize() local
184 const Tensor& qtensor, in per_tensor_affine_dequantize_impl()
194 Tensor& rtensor, const Tensor& qtensor) { in dequantize_out()
207 Tensor PerTensorAffineQuantizer::dequantize(const Tensor& qtensor) { in dequantize()
220 Tensor qtensor = new_qtensor( in quantize() local
234 const Tensor& qtensor, in per_channel_affine_dequantize_impl()
244 Tensor PerChannelAffineQuantizer::dequantize(const Tensor& qtensor) { in dequantize()
255 Tensor& rtensor, const Tensor& qtensor) { in dequantize_out()
272 Tensor qtensor = new_qtensor( in quantize() local
284 const Tensor& qtensor, in per_channel_affine_float_q_params_dequantize_impl()
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/
H A DAffineQuantizer.cu42 Tensor& qtensor, in quantize_tensor_per_tensor_affine_cuda()
70 const Tensor& qtensor, in dequantize_tensor_per_tensor_affine_cuda()
89 Tensor& qtensor, in quantize_tensor_per_channel_affine_cuda()
131 const Tensor& qtensor, in dequantize_tensor_per_channel_affine_cuda()
168 Tensor& qtensor, in quantize_tensor_per_channel_float_qparams_cuda()
215 const Tensor& qtensor, in dequantize_tensor_per_channel_float_qparams_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DAffineQuantizer.cpp111 Tensor& qtensor, in quantize_tensor_per_tensor_affine()
140 Tensor& qtensor, in quantize_tensor_per_channel_affine()
175 Tensor& qtensor, in quantize_tensor_per_channel_float_qparams()
206 const Tensor& qtensor, in dequantize_tensor_per_tensor_affine()
231 const Tensor& qtensor, in dequantize_tensor_per_channel_affine()
265 const Tensor& qtensor, in dequantize_tensor_per_channel_float_qparams()
H A DTensorFactories.cpp135 const Tensor& qtensor, in empty_quantized()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp3318 Tensor& qtensor,
3350 const Tensor& qtensor,
3652 Tensor& qtensor,
3685 const Tensor& qtensor,
3724 Tensor& qtensor,
3783 Tensor& qtensor,
3969 Tensor& qtensor,
3987 const Tensor& qtensor,
4051 const Tensor& qtensor,
4065 Tensor& qtensor,
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dquantized_test.cpp237 …Tensor qtensor = at::from_blob_quantized_per_tensor_affine(custom_data, shape, deleter, scale, zer… in TEST() local
280 …Tensor qtensor = at::from_blob_quantized_per_channel_affine(custom_data, shape, deleter, scales, z… in TEST() local
/aosp_15_r20/external/executorch/backends/xnnpack/runtime/utils/
H A Dutils.h131 executorch::aten::Tensor& qtensor, in QuantizePerTensor()
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp124 at::Tensor& qtensor, in quantize_tensor_per_tensor_affine_privateuse1()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_prim_ops.cpp906 at::Tensor qtensor; in __anonbfe5918f3b02() local