burn::tensor::quantization

Trait QTensorPrimitive

Source 
pub trait QTensorPrimitive {
    // Required method
    fn scheme(&self) -> &QuantizationScheme;
}
Expand description

Quantized tensor primitive.

Required Methods§

Source

fn scheme(&self) -> &QuantizationScheme

Returns the quantization scheme for the given tensor.

Implementors§