Struct QParams

pub struct QParams { /* private fields */ }

Quantization parameters.

Implementations

impl QParams

pub fn new(scheme: QuantizationScheme) -> Self

Create a new quantization parameters instance.

pub fn values(&self, tensor: &QTensor) -> (f32, i32)

Get the quantization parameters values.

pub fn __expand_new( context: &mut CubeContext, scheme: <QuantizationScheme as CubeType>::ExpandType, ) -> <Self as CubeType>::ExpandType

pub fn __expand_values( context: &mut CubeContext, this: <Self as CubeType>::ExpandType, tensor: <QTensor as CubeType>::ExpandType, ) -> <(f32, i32) as CubeType>::ExpandType

Trait Implementations

impl CubeType for QParams

type ExpandType = QParamsExpand

fn init(context: &mut CubeContext, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl IntoRuntime for QParams

fn __expand_runtime_method(self, context: &mut CubeContext) -> Self::ExpandType

fn runtime(self) -> Self

Make sure a type is actually expanded into its runtime expand type.

impl LaunchArg for QParams

type RuntimeArg<'a, R: Runtime> = QParamsLaunch<'a, R>

The runtime argument for the kernel.

fn compilation_arg<'a, R: Runtime>( runtime_arg: &Self::RuntimeArg<'a, R>, ) -> Self::CompilationArg

impl LaunchArgExpand for QParams

type CompilationArg = QParamsCompilationArg

Compilation argument.

fn expand( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register an input variable during compilation that fill the KernelBuilder.

fn expand_output( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register an output variable during compilation that fill the KernelBuilder.