Struct TensorArg 

pub struct TensorArg<P: ReduceDType, RA: ReduceArgs, Tag> { /* private fields */ }

Implementations

impl<P: ReduceDType, RA: ReduceArgs> TensorArg<P, RA, Input>

pub fn new_input(_state: &RA::State<P>) -> Self

pub fn __expand_new_input( _scope: &mut Scope, state: <RA::State<P> as CubeType>::ExpandType, ) -> TensorArgExpand<P, RA, Input>

impl<P: ReduceDType, RA: ReduceArgs> TensorArg<P, RA, Output>

pub fn new_output(_state: &mut RA::State<P>) -> Self

pub fn __expand_new_output( _scope: &mut Scope, state: <RA::State<P> as CubeType>::ExpandType, ) -> TensorArgExpand<P, RA, Output>

Trait Implementations

impl<P: ReduceDType, RA: ReduceArgs, Tag> CubeType for TensorArg<P, RA, Tag>

type ExpandType = TensorArgExpand<P, RA, Tag>

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

Wrapper around the init method, necessary to type inference.

impl<P: ReduceDType, RA: ReduceArgs> Lined for TensorArg<P, RA, Input>

fn line_size(&self) -> u32

fn __expand_line_size(_scope: &mut Scope, this: Self::ExpandType) -> u32

impl<P: ReduceDType, RA: ReduceArgs> Lined for TensorArg<P, RA, Output>

fn line_size(&self) -> u32

fn __expand_line_size(_scope: &mut Scope, this: Self::ExpandType) -> u32

impl<P: ReduceDType, RA: ReduceArgs> VirtualTensorOperations<<P as ReduceDType>::In> for TensorArg<P, RA, Input>

fn as_tensor_map(&self) -> CubeOption<TensorMap<E>>

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

Read the tensor at the given index.

fn read_window(&self, _start: u32, _end: u32) -> Slice<Line<E>>

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.

fn len(&self) -> u32

fn buffer_len(&self) -> u32

fn __expand_as_tensor_map( scope: &mut Scope, this: Self::ExpandType, ) -> <CubeOption<TensorMap<E>> as CubeType>::ExpandType

fn __expand_read( scope: &mut Scope, this: Self::ExpandType, _index: <u32 as CubeType>::ExpandType, ) -> <Line<E> as CubeType>::ExpandType

fn __expand_read_window( scope: &mut Scope, this: Self::ExpandType, _start: <u32 as CubeType>::ExpandType, _end: <u32 as CubeType>::ExpandType, ) -> <Slice<Line<E>> as CubeType>::ExpandType

fn __expand_write( scope: &mut Scope, this: Self::ExpandType, _index: <u32 as CubeType>::ExpandType, _value: <Line<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

fn __expand_shape( scope: &mut Scope, this: Self::ExpandType, _axis: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

fn __expand_stride( scope: &mut Scope, this: Self::ExpandType, _axis: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

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

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

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

impl<P: ReduceDType, RA: ReduceArgs> VirtualTensorOperations<<P as ReduceDType>::Out> for TensorArg<P, RA, Output>

fn as_tensor_map(&self) -> CubeOption<TensorMap<E>>

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

Read the tensor at the given index.

fn read_window(&self, _start: u32, _end: u32) -> Slice<Line<E>>

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.

fn len(&self) -> u32

fn buffer_len(&self) -> u32

fn __expand_as_tensor_map( scope: &mut Scope, this: Self::ExpandType, ) -> <CubeOption<TensorMap<E>> as CubeType>::ExpandType

fn __expand_read( scope: &mut Scope, this: Self::ExpandType, _index: <u32 as CubeType>::ExpandType, ) -> <Line<E> as CubeType>::ExpandType

fn __expand_read_window( scope: &mut Scope, this: Self::ExpandType, _start: <u32 as CubeType>::ExpandType, _end: <u32 as CubeType>::ExpandType, ) -> <Slice<Line<E>> as CubeType>::ExpandType

fn __expand_write( scope: &mut Scope, this: Self::ExpandType, _index: <u32 as CubeType>::ExpandType, _value: <Line<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

fn __expand_shape( scope: &mut Scope, this: Self::ExpandType, _axis: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

fn __expand_stride( scope: &mut Scope, this: Self::ExpandType, _axis: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

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

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

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