Struct TensorOutputExpand 

pub struct TensorOutputExpand<Q: FloatLine, K: FloatLine, V: FloatLine, M: NumericLine, O: FloatLine, GA: AttentionArgs> { /* private fields */ }

Expand type for tensor output.

Implementations

impl<Q: FloatLine, K: FloatLine, V: FloatLine, M: NumericLine, O: FloatLine, GA: AttentionArgs> TensorOutputExpand<Q, K, V, M, O, GA>

pub fn __expand_write_method( self, scope: &mut Scope, coordinate: <usize as RudaType>::ExpandType, val: <Vector<O::T, O::N> as RudaType>::ExpandType, ) -> <() as RudaType>::ExpandType

pub fn __expand_shape_method( self, scope: &mut Scope, axis: <usize as RudaType>::ExpandType, ) -> <usize as RudaType>::ExpandType

pub fn __expand_stride_method( self, scope: &mut Scope, dim: <usize as RudaType>::ExpandType, ) -> <usize as RudaType>::ExpandType

pub fn __expand_rank_method( self, scope: &mut Scope, ) -> <usize as RudaType>::ExpandType

pub fn __expand_len_method( self, scope: &mut Scope, ) -> <usize as RudaType>::ExpandType

pub fn __expand_buffer_len_method( self, scope: &mut Scope, ) -> <usize as RudaType>::ExpandType

pub fn __expand_vector_size_method(self, scope: &mut Scope) -> VectorSize

pub fn __expand_new( scope: &mut Scope, state: <GA::State<Q, K, V, M, O> as RudaType>::ExpandType, ) -> <TensorOutput<Q, K, V, M, O, GA> as RudaType>::ExpandType

pub fn __expand_write( scope: &mut Scope, this: Self, coordinate: <usize as RudaType>::ExpandType, val: <Vector<O::T, O::N> as RudaType>::ExpandType, ) -> <() as RudaType>::ExpandType

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

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

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

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

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

pub fn __expand_vector_size(scope: &mut Scope, this: Self) -> VectorSize

Trait Implementations

impl<Q: FloatLine, K: FloatLine, V: FloatLine, M: NumericLine, O: FloatLine, GA: AttentionArgs> Clone for TensorOutputExpand<Q, K, V, M, O, GA>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<Q: FloatLine, K: FloatLine, V: FloatLine, M: NumericLine, O: FloatLine, GA: AttentionArgs> IntoMut for TensorOutputExpand<Q, K, V, M, O, GA>

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

Convert the variable into a potentially new mutable variable in scope, copying if needed.

fn expand_inactive_value(self, _scope: &mut Scope) -> Self

fn expand_reference(self, _scope: &mut Scope, _writable: bool) -> Self

fn expand_absent_reference(self, _scope: &mut Scope, _writable: bool) -> Self

fn expand_reference_value(self, _scope: &mut Scope) -> Self

fn expand_reborrow(self, _scope: &mut Scope, _writable: bool) -> Self

fn expand_return_reference(self, scope: &mut Scope, writable: bool) -> Self

fn expand_deref(self, _scope: &mut Scope) -> Self

impl<Q: FloatLine, K: FloatLine, V: FloatLine, M: NumericLine, O: FloatLine, GA: AttentionArgs> RudaDebug for TensorOutputExpand<Q, K, V, M, O, GA>

fn set_debug_name(&self, scope: &mut Scope, name: &'static str)

Set the debug name of this type’s expansion. Should do nothing for types that don’t appear at runtime

impl<Q: FloatLine, K: FloatLine, V: FloatLine, M: NumericLine, O: FloatLine, MA: AttentionArgs> VectorizedExpand for TensorOutputExpand<Q, K, V, M, O, MA>

fn vector_size(&self) -> VectorSize

fn __expand_vector_size_method(&self, _scope: &mut Scope) -> usize

impl<Q: FloatLine, K: FloatLine, V: FloatLine, M: NumericLine, O: FloatLine, MA: AttentionArgs> VirtualTensorOperationsExpand<<O as FloatLine>::T, <O as FloatLine>::N> for TensorOutputExpand<Q, K, V, M, O, MA>

fn __expand_read_method( &self, _scope: &mut Scope, _index: NativeExpand<usize>, ) -> NativeExpand<Vector<O::T, O::N>>

fn __expand_read_window_method( &self, _context: &mut Scope, _start: NativeExpand<usize>, _end: NativeExpand<usize>, ) -> SliceExpand<Vector<O::T, O::N>, ReadOnly>

fn __expand_write_method( &self, scope: &mut Scope, index: NativeExpand<usize>, val: NativeExpand<Vector<O::T, O::N>>, )

fn __expand_shape_method( &self, scope: &mut Scope, axis: NativeExpand<usize>, ) -> NativeExpand<usize>

fn __expand_stride_method( &self, scope: &mut Scope, axis: NativeExpand<usize>, ) -> NativeExpand<usize>

fn __expand_rank_method(&self, scope: &mut Scope) -> NativeExpand<usize>

fn __expand_len_method(&self, scope: &mut Scope) -> NativeExpand<usize>

fn __expand_buffer_len_method(&self, scope: &mut Scope) -> NativeExpand<usize>

fn __expand_as_tensor_map_method( &self, scope: &mut Scope, ) -> ComptimeOptionExpand<TensorMap<O::T, Tiled>>