Struct StridedLayoutLaunch 

pub struct StridedLayoutLaunch<'a, R: Runtime> { /* private fields */ }

Implementations

impl<'a, R: Runtime> StridedLayoutLaunch<'a, R>

pub fn new( shape: <FastDivmod as LaunchArg>::RuntimeArg<'a, R>, stride: <u32 as LaunchArg>::RuntimeArg<'a, R>, len: <u32 as LaunchArg>::RuntimeArg<'a, R>, line_size: u32, ) -> Self

New kernel

impl<'a, R: Runtime> StridedLayoutLaunch<'a, R>

pub fn from_shape_strides( client: &ComputeClient<R::Server>, shape: &[usize], strides: &[usize], line_size: u8, ) -> Self

pub fn from_handle( client: &ComputeClient<R::Server>, handle: &TensorHandleRef<'_, R>, line_size: u8, ) -> Self

Trait Implementations

impl<'a, R: Runtime> ArgSettings<R> for StridedLayoutLaunch<'a, R>

fn register(&self, launcher: &mut KernelLauncher<R>)

Register the information of an argument to the KernelLauncher.