Enum InputBinding 

pub enum InputBinding<R: Runtime> {
    Normal(TensorBinding<R>, StorageType),
    Quantized {
        data: TensorBinding<R>,
        data_dtype: StorageType,
        scale: TensorBinding<R>,
        scale_dtype: StorageType,
        shape: Shape,
        scheme: QuantScheme,
    },
}

Variants

Normal(TensorBinding<R>, StorageType)

Quantized

Fields

data: TensorBinding<R>

data_dtype: StorageType

scale: TensorBinding<R>

scale_dtype: StorageType

shape: Shape

Unpacked shape, excluding padding

scheme: QuantScheme

Implementations

impl<R: Runtime> InputBinding<R>

pub fn new(data: TensorBinding<R>, dtype: StorageType) -> Self

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

pub fn quantized( data: TensorBinding<R>, scale: TensorBinding<R>, shape: Shape, scheme: QuantScheme, data_dtype: StorageType, scale_dtype: StorageType, ) -> Self

pub fn data(&self) -> &TensorBinding<R>

pub fn data_elem_size(&self) -> usize

pub fn into_data(self) -> TensorBinding<R>

pub fn data_mut(&mut self) -> &mut TensorBinding<R>

pub fn scale(&self) -> Option<&TensorBinding<R>>

pub fn scheme(&self) -> Option<&QuantScheme>

pub fn shape(&self) -> &Shape

pub fn into_contiguous( self, client: &ComputeClient<R>, ) -> Result<Self, LaunchError>

pub fn required_address_type(&self) -> AddressType

Trait Implementations

impl<R: Runtime> Clone for InputBinding<R>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<R: Debug + Runtime> Debug for InputBinding<R>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.