Trait burn_tensor::quantization::QTensorPrimitive

pub trait QTensorPrimitive {
    // Required methods
    fn scheme(&self) -> &QuantizationScheme;
    fn strategy(&self) -> QuantizationStrategy;
}

Quantized tensor primitive.

Required Methods

fn scheme(&self) -> &QuantizationScheme

Returns the quantization scheme for the given tensor.

fn strategy(&self) -> QuantizationStrategy

Returns the quantization strategy for the given tensor.

Remarks

Retrieving the quantization strategy with its corresponding parameters might require synchronization on the backend.

Implementors