Enum MatmulInputHandle 

pub enum MatmulInputHandle<R: Runtime, E: CubePrimitive, S: CubePrimitive = f32> {
    Normal(TensorHandle<R, E>),
    Quantized {
        data: TensorHandle<R, E>,
        scale: TensorHandle<R, S>,
        shape: Vec<usize>,
        scheme: QuantScheme,
    },
}

Variants

Normal(TensorHandle<R, E>)

Quantized

Fields

data: TensorHandle<R, E>

scale: TensorHandle<R, S>

shape: Vec<usize>

scheme: QuantScheme

Implementations

impl<R: Runtime, E: Numeric> MatmulInputHandle<R, E>

pub fn as_ref(&self) -> MatmulInputHandleRef<'_, R>

pub fn from_ref(handle: &MatmulInputHandleRef<'_, R>) -> Self

pub fn data(&self) -> &TensorHandle<R, E>

pub fn swap_dims(&mut self, dim0: usize, dim1: usize)

Trait Implementations

impl<R: Runtime, E: CubePrimitive> Clone for MatmulInputHandle<R, E>

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.