Struct TensorQueryExpand 

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

Expand type for tensor input.

Implementations

impl<Q: FloatLine, K: FloatLine, V: FloatLine, M: NumericLine, O: FloatLine, MA: AttentionArgs> TensorQueryExpand<Q, K, V, M, O, MA>

pub fn __expand_read_window_method( self, scope: &mut Scope, start: <usize as RudaType>::ExpandType, end: <usize as RudaType>::ExpandType, ) -> <Slice<Vector<Q::T, Q::N>> as RudaType>::ExpandType

pub fn __expand_read_method( self, scope: &mut Scope, coordinate: <usize as RudaType>::ExpandType, ) -> <Vector<Q::T, Q::N> 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, axis: <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_as_tensor_map_method( self, scope: &mut Scope, ) -> <ComptimeOption<TensorMap<Q::T, Tiled>> as RudaType>::ExpandType

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

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

pub fn __expand_read_window( scope: &mut Scope, this: Self, start: <usize as RudaType>::ExpandType, end: <usize as RudaType>::ExpandType, ) -> <Slice<Vector<Q::T, Q::N>> as RudaType>::ExpandType

pub fn __expand_read( scope: &mut Scope, this: Self, coordinate: <usize as RudaType>::ExpandType, ) -> <Vector<Q::T, Q::N> 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, axis: <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_as_tensor_map( scope: &mut Scope, this: Self, ) -> <ComptimeOption<TensorMap<Q::T, Tiled>> 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 TensorQueryExpand<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 TensorQueryExpand<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 TensorQueryExpand<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 TensorQueryExpand<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<<Q as FloatLine>::T, <Q as FloatLine>::N> for TensorQueryExpand<Q, K, V, M, O, MA>

fn __expand_read_method( &self, scope: &mut Scope, index: NativeExpand<usize>, ) -> NativeExpand<Vector<Q::T, Q::N>>

fn __expand_read_window_method( &self, context: &mut Scope, start: NativeExpand<usize>, end: NativeExpand<usize>, ) -> SliceExpand<Vector<Q::T, Q::N>, ReadOnly>

fn __expand_write_method( &self, _scope: &mut Scope, _index: NativeExpand<usize>, _val: NativeExpand<Vector<Q::T, Q::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<Q::T, Tiled>>