Enum LinearLayoutArgs 

pub enum LinearLayoutArgs<'a, R: Runtime> {
    Plain(<PlainLayout as LaunchArg>::RuntimeArg<'a, R>),
    Strided(<StridedLayout as LaunchArg>::RuntimeArg<'a, R>),
    Permuted(<PermutedLayout as LaunchArg>::RuntimeArg<'a, R>),
}

Variants

Plain(<PlainLayout as LaunchArg>::RuntimeArg<'a, R>)

Strided(<StridedLayout as LaunchArg>::RuntimeArg<'a, R>)

Permuted(<PermutedLayout as LaunchArg>::RuntimeArg<'a, R>)

Implementations

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

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

Construct a linear layout from shapes, strides and line size of the tensor

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

Construct a possibly broadcast linear layout from shapes/strides and a reference shape

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

Construct a linear layout from a tensor handle

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

Construct a possibly broadcast linear layout from a tensor handle and reference handle

Trait Implementations

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

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

Register the information of an argument to the KernelLauncher.