Skip to main content

build_zero_point_tensor

quantize_rs::onnx_utils::quantization_nodes

Function build_zero_point_tensor 

Source 
pub fn build_zero_point_tensor(
    names: &DequantLinearNames,
    zps: &[i8],
) -> TensorProto
Expand description

INT8 zero-point tensor.

For per-tensor, zps has one element → rank-0 scalar. For per-channel, zps has one per channel → rank-1 [num_channels].