Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/quantized/
H A DQuantizer.h79 struct TORCH_API PerTensorAffineQuantizer : public AffineQuantizer { struct
80 explicit PerTensorAffineQuantizer(ScalarType scalar_type, double scale, int64_t zero_point) in PerTensorAffineQuantizer() argument
106 static_cast<PerTensorAffineQuantizer*>(other.get()); in equalTo() argument
H A DQuantizer.cpp44 return c10::make_intrusive<PerTensorAffineQuantizer>(scalar_type, in make_per_tensor_affine_quantizer()
163 Tensor PerTensorAffineQuantizer::quantize(const Tensor& rtensor) { in quantize()
193 Tensor& PerTensorAffineQuantizer::dequantize_out( in dequantize_out()
207 Tensor PerTensorAffineQuantizer::dequantize(const Tensor& qtensor) { in dequantize()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DQTensor.cpp120 return static_cast<PerTensorAffineQuantizer*>(quantizer.get())->scale(); in q_scale_quant()
126 return static_cast<PerTensorAffineQuantizer*>(quantizer.get())->zero_point(); in q_zero_point_quant()