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 RudaType>::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 RudaType>::ExpandType, ) -> TensorArgExpand<P, RA, Output>

Trait Implementations

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

type ExpandType = TensorArgExpand<P, RA, Tag>

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

fn vector_size(&self) -> usize

fn __expand_vector_size(_scope: &mut Scope, this: Self::ExpandType) -> usize

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

fn vector_size(&self) -> usize

fn __expand_vector_size(_scope: &mut Scope, this: Self::ExpandType) -> usize

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

fn as_tensor_map(&self) -> ComptimeOption<TensorMap<E, Tiled>>

fn read(&self, _index: usize) -> Vector<E, N>

Read the tensor at the given index.

fn read_window(&self, _start: usize, _end: usize) -> Slice<Vector<E, N>>

fn write(&self, _index: usize, _value: Vector<E, N>)

Write the tensor at the given index.

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

Get the shape of the tensor at the given axis.

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

Get the stride of the tensor at the given axis.

fn rank(&self) -> usize

Get the rank of the tensor.

fn len(&self) -> usize

fn buffer_len(&self) -> usize

fn __expand_as_tensor_map( scope: &mut Scope, this: Self::ExpandType, ) -> <ComptimeOption<TensorMap<E, Tiled>> as RudaType>::ExpandType

fn __expand_read( scope: &mut Scope, this: Self::ExpandType, _index: <usize as RudaType>::ExpandType, ) -> <Vector<E, N> as RudaType>::ExpandType

fn __expand_read_window( scope: &mut Scope, this: Self::ExpandType, _start: <usize as RudaType>::ExpandType, _end: <usize as RudaType>::ExpandType, ) -> <Slice<Vector<E, N>> as RudaType>::ExpandType

fn __expand_write( scope: &mut Scope, this: Self::ExpandType, _index: <usize as RudaType>::ExpandType, _value: <Vector<E, N> as RudaType>::ExpandType, ) -> <() as RudaType>::ExpandType

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

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

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

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

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

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

fn as_tensor_map(&self) -> ComptimeOption<TensorMap<E, Tiled>>

fn read(&self, _index: usize) -> Vector<E, N>

Read the tensor at the given index.

fn read_window(&self, _start: usize, _end: usize) -> Slice<Vector<E, N>>

fn write(&self, _index: usize, _value: Vector<E, N>)

Write the tensor at the given index.

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

Get the shape of the tensor at the given axis.

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

Get the stride of the tensor at the given axis.

fn rank(&self) -> usize

Get the rank of the tensor.

fn len(&self) -> usize

fn buffer_len(&self) -> usize

fn __expand_as_tensor_map( scope: &mut Scope, this: Self::ExpandType, ) -> <ComptimeOption<TensorMap<E, Tiled>> as RudaType>::ExpandType

fn __expand_read( scope: &mut Scope, this: Self::ExpandType, _index: <usize as RudaType>::ExpandType, ) -> <Vector<E, N> as RudaType>::ExpandType

fn __expand_read_window( scope: &mut Scope, this: Self::ExpandType, _start: <usize as RudaType>::ExpandType, _end: <usize as RudaType>::ExpandType, ) -> <Slice<Vector<E, N>> as RudaType>::ExpandType

fn __expand_write( scope: &mut Scope, this: Self::ExpandType, _index: <usize as RudaType>::ExpandType, _value: <Vector<E, N> as RudaType>::ExpandType, ) -> <() as RudaType>::ExpandType

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

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

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

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

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