models/quantize_affine.py [120:132]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        input: torch.Tensor,
        quantize_params: QuantizeAffineParams2,
    ) -> torch.Tensor:
        quantized_tensor = get_quantized_representation(input, quantize_params)
        dequantized_tensor = dequantize(quantized_tensor, quantize_params)

        mark_quantize_affine(
            dequantized_tensor,
            quantize_params.scale,
            quantize_params.zero_point,
            quantize_params.num_bits,
        )
        return dequantized_tensor
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



models/quantize_affine.py [141:153]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    input: torch.Tensor,
    quantize_params: QuantizeAffineParams2,
) -> torch.Tensor:
    quantized_tensor = get_quantized_representation(input, quantize_params)
    dequantized_tensor = dequantize(quantized_tensor, quantize_params)

    mark_quantize_affine(
        dequantized_tensor,
        quantize_params.scale,
        quantize_params.zero_point,
        quantize_params.num_bits,
    )
    return dequantized_tensor
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -