Struct TensorView 

pub struct TensorView<T: CubePrimitive> { /* private fields */ }

Implementations

impl<T: CubePrimitive> TensorView<T>

pub fn new(buffer: &[T], shape: Sequence<u32>) -> TensorViewBuilder<T>

pub fn slice( &self, offsets: Sequence<u32>, shape: Sequence<u32>, ) -> TensorView<T>

pub fn __expand_new( scope: &Scope, buffer: &<[T] as CubeType>::ExpandType, shape: <Sequence<u32> as CubeType>::ExpandType, ) -> <TensorViewBuilder<T> as CubeType>::ExpandType

pub fn __expand_slice( scope: &Scope, this: &<Self as CubeType>::ExpandType, offsets: <Sequence<u32> as CubeType>::ExpandType, shape: <Sequence<u32> as CubeType>::ExpandType, ) -> <TensorView<T> as CubeType>::ExpandType

impl<T: CubePrimitive> TensorView<T>

pub fn permuted(&self, _permutation: Sequence<usize>) -> TensorView<T>

Trait Implementations

impl<T: Clone + CubePrimitive> Clone for TensorView<T>

fn clone(&self) -> TensorView<T>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T: CubePrimitive> CubeType for TensorView<T>

type ExpandType = TensorViewExpand<T>

impl<T: CubePrimitive> LaunchArg for TensorView<T>

type RuntimeArg<R: Runtime> = TensorViewBuilderLaunch<T, R>

The runtime argument for the kernel.

type CompilationArg = TensorViewBuilderCompilationArg<T>

Compilation argument.

fn register<R: Runtime>( arg: Self::RuntimeArg<R>, launcher: &mut KernelLauncher<R>, ) -> Self::CompilationArg

fn expand( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register a variable during compilation that fill the KernelBuilder.