Struct TensorOutput

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

Tensor output representation.

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

Warning

There is no mutability guarantee.

Implementations

impl<EI: Numeric, EO: Numeric, GA: MatmulArgs> TensorOutput<EI, EO, GA>

pub fn new(state: &mut GA::State<EI, EO>) -> TensorOutput<EI, EO, GA>

Create a tensor output from the state.

pub fn write(&self, coordinate: u32, value: Line<EO>)

Write the value to 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, dim: 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 __expand_new( context: &mut Scope, state: <GA::State<EI, EO> as CubeType>::ExpandType, ) -> <TensorOutput<EI, EO, GA> as CubeType>::ExpandType

pub fn __expand_write( context: &mut Scope, this: <Self as CubeType>::ExpandType, coordinate: <u32 as CubeType>::ExpandType, value: <Line<EO> as CubeType>::ExpandType, ) -> <() 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, dim: <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

Trait Implementations

impl<EI: Numeric, EO: Numeric, GA: MatmulArgs> Clone for TensorOutput<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 TensorOutput<EI, EO, GA>

type ExpandType = TensorOutputExpand<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<EO> for TensorOutput<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 TensorOutput<EI, EO, GA>