Trait burn_core::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.

Implementations on Foreign Types

impl<Q, const D: usize> QTensorPrimitive for NdArrayQTensor<Q, D>

where Q: QuantElement,

fn scheme(&self) -> &QuantizationScheme

fn strategy(&self) -> QuantizationStrategy

impl<Q, const D: usize> QTensorPrimitive for TchQTensor<Q, D>

where Q: QuantElement,

fn scheme(&self) -> &QuantizationScheme

fn strategy(&self) -> QuantizationStrategy

impl<R> QTensorPrimitive for QFusionTensor<R>

where R: FusionRuntime,

fn scheme(&self) -> &QuantizationScheme

fn strategy(&self) -> QuantizationStrategy

impl<const D: usize> QTensorPrimitive for CandleQTensor<D>

fn scheme(&self) -> &QuantizationScheme

fn strategy(&self) -> QuantizationStrategy

Implementors