Enum InputBinding 

pub enum InputBinding {
    Normal(TensorBinding, ElemType),
    Quantized {
        data: TensorBinding,
        data_dtype: ElemType,
        scale: TensorBinding,
        scale_dtype: ElemType,
        shape: Shape,
        scheme: QuantScheme,
    },
}

Variants

Normal(TensorBinding, ElemType)

Quantized

Fields

data: TensorBinding

data_dtype: ElemType

scale: TensorBinding

scale_dtype: ElemType

shape: Shape

Unpacked shape, excluding padding

scheme: QuantScheme

Implementations

impl InputBinding

pub fn new(data: TensorBinding, dtype: ElemType) -> Self

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

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

pub fn data(&self) -> &TensorBinding

pub fn data_elem_size(&self) -> usize

pub fn into_data(self) -> TensorBinding

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

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

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

pub fn shape(&self) -> &Shape

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

pub fn required_address_type(&self) -> AddressType

Trait Implementations

impl Clone for InputBinding

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for InputBinding

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

Formats the value using the given formatter.