Searched refs:PerTensorAffineQuantizer (Results 1 – 3 of 3) sorted by relevance
79 struct TORCH_API PerTensorAffineQuantizer : public AffineQuantizer { struct80 explicit PerTensorAffineQuantizer(ScalarType scalar_type, double scale, int64_t zero_point) in PerTensorAffineQuantizer() argument106 static_cast<PerTensorAffineQuantizer*>(other.get()); in equalTo() argument
44 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()
120 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()