Struct TensorInput

pub struct TensorInput<EI: Numeric, EO: Numeric, GA: MatmulArgs> { /* private fields */ }

Tensor input representation.

You can use the tensor input as if it was a pointer to the actually tensor.

Implementations

impl<EI: Numeric, EO: Numeric, MA: MatmulArgs> TensorInput<EI, EO, MA>

pub fn new( state: &MA::State<EI, EO>, ident: TensorInputIdent, ) -> TensorInput<EI, EO, MA>

Create a tensor input from the state and the ident.

pub fn read_window(&self, start: u32, end: u32) -> Slice<Line<EI>>

pub fn read(&self, coordinate: u32) -> Line<EI>

Read the tensor at the given coordinate.

pub fn shape(&self, axis: u32) -> u32

Get the shape of the tensor at the given axis.

pub fn stride(&self, axis: u32) -> u32

Get the stride of the tensor at the given axis.

pub fn rank(&self) -> u32

Get the rank of the tensor.

pub fn len(&self) -> u32

Get the length of the tensor.

pub fn buffer_len(&self) -> u32

Get the buffer length of the tensor.

pub fn as_tensor_map(&self) -> TensorMap<EI>

Get the buffer length of the tensor.

pub fn __expand_new( context: &mut Scope, state: <MA::State<EI, EO> as CubeType>::ExpandType, ident: TensorInputIdent, ) -> <TensorInput<EI, EO, MA> as CubeType>::ExpandType

pub fn __expand_read_window( context: &mut Scope, this: <Self as CubeType>::ExpandType, start: <u32 as CubeType>::ExpandType, end: <u32 as CubeType>::ExpandType, ) -> <Slice<Line<EI>> as CubeType>::ExpandType

pub fn __expand_read( context: &mut Scope, this: <Self as CubeType>::ExpandType, coordinate: <u32 as CubeType>::ExpandType, ) -> <Line<EI> as CubeType>::ExpandType

pub fn __expand_shape( context: &mut Scope, this: <Self as CubeType>::ExpandType, axis: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_stride( context: &mut Scope, this: <Self as CubeType>::ExpandType, axis: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_rank( context: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_len( context: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_buffer_len( context: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_as_tensor_map( context: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <TensorMap<EI> as CubeType>::ExpandType

Trait Implementations

impl<EI: Numeric, EO: Numeric, GA: MatmulArgs> Clone for TensorInput<EI, EO, GA>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<EI: Numeric, EO: Numeric, GA: MatmulArgs> CubeType for TensorInput<EI, EO, GA>

type ExpandType = TensorInputExpand<EI, EO, GA>

fn init(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl<EI: Numeric, EO: Numeric, MA: MatmulArgs> VirtualTensorOperations<EI> for TensorInput<EI, EO, MA>

fn read(&self, _index: u32) -> Line<E>

Read the tensor at the given index.

fn write(&self, _index: u32, _value: Line<E>)

Write the tensor at the given index.

fn shape(&self, _axis: u32) -> u32

Get the shape of the tensor at the given axis.

fn stride(&self, _axis: u32) -> u32

Get the stride of the tensor at the given axis.

fn rank(&self) -> u32

Get the rank of the tensor.

impl<EI: Numeric, EO: Numeric, GA: MatmulArgs> Copy for TensorInput<EI, EO, GA>