Skip to main content

params_shape

Function params_shape 

Source
pub fn params_shape(data_shape: &Shape, scheme: &QuantScheme) -> Shape
Expand description

Calculate the shape of the block scale grid for a given tensor and scheme.

A two-level scheme’s per-tensor scale is not part of this grid.