Struct PermutedLayoutLaunch 

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

Implementations

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

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

New kernel

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

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

Create a new permuted layout for a possibly broadcast tensor, with a reference shape to be broadcast to.

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

Create a new permuted layout for a possibly broadcast tensor, with a reference shape to be broadcast to.

pub fn from_handles_ref( client: &ComputeClient<R::Server>, handle: &TensorHandleRef<'_, R>, reference_handle: &TensorHandleRef<'_, R>, 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 PermutedLayoutLaunch<'a, R>

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

Register the information of an argument to the KernelLauncher.