burn_cubecl::kernel::quantization

Function quantize

Source 
pub fn quantize<R, F, I>(
    tensor: CubeTensor<R>,
    scheme: &QuantizationScheme,
    scale: CubeTensor<R>,
) -> CubeTensor<R>
where
    R: CubeRuntime,
    F: CubeElement,
    I: IntElement,
Expand description

Convert the tensor to a lower precision data type based on the quantization scheme and parameters.