Skip to main content

quantize_tensor

Function quantize_tensor 

Source
pub fn quantize_tensor(
    values: &[f32],
    shape: &[usize],
    granularity: QuantGranularity,
    mode: QuantMode,
    bits: u8,
) -> QuantizedTensor
Expand description

Quantize tensor with specified granularity

ยงArguments

  • values - Input tensor values
  • shape - Tensor shape
  • granularity - Quantization granularity
  • mode - Quantization mode
  • bits - Bit width (4 or 8)