burn_tensor::quantization

Trait QTensorPrimitive

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

Quantized tensor primitive.

Required Methods§

Source

fn scheme(&self) -> &QuantizationScheme

Returns the quantization scheme for the given tensor.

Source

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§