Struct NhwcLayoutLaunch 

pub struct NhwcLayoutLaunch<'a, R: Runtime> {
    pub stride_batch: <u32 as LaunchArg>::RuntimeArg<'a, R>,
    pub strides_spatial: <Sequence<u32> as LaunchArg>::RuntimeArg<'a, R>,
    pub stride_channel: <u32 as LaunchArg>::RuntimeArg<'a, R>,
    pub shape_batch: <u32 as LaunchArg>::RuntimeArg<'a, R>,
    pub shapes_spatial: <Sequence<u32> as LaunchArg>::RuntimeArg<'a, R>,
    pub shape_channel: <u32 as LaunchArg>::RuntimeArg<'a, R>,
    pub line_size: u32,
    pub check_spatial: bool,
    /* private fields */
}

Fields

stride_batch: <u32 as LaunchArg>::RuntimeArg<'a, R>

strides_spatial: <Sequence<u32> as LaunchArg>::RuntimeArg<'a, R>

stride_channel: <u32 as LaunchArg>::RuntimeArg<'a, R>

shape_batch: <u32 as LaunchArg>::RuntimeArg<'a, R>

shapes_spatial: <Sequence<u32> as LaunchArg>::RuntimeArg<'a, R>

shape_channel: <u32 as LaunchArg>::RuntimeArg<'a, R>

line_size: u32

check_spatial: bool

Implementations

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

pub fn new( stride_batch: <u32 as LaunchArg>::RuntimeArg<'a, R>, strides_spatial: <Sequence<u32> as LaunchArg>::RuntimeArg<'a, R>, stride_channel: <u32 as LaunchArg>::RuntimeArg<'a, R>, shape_batch: <u32 as LaunchArg>::RuntimeArg<'a, R>, shapes_spatial: <Sequence<u32> as LaunchArg>::RuntimeArg<'a, R>, shape_channel: <u32 as LaunchArg>::RuntimeArg<'a, R>, line_size: u32, check_spatial: bool, ) -> Self

New kernel

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

pub fn from_handle( handle: &TensorHandleRef<'a, R>, line_size: u32, check_spatial: bool, ) -> Self

Trait Implementations

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

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

Register the information of an argument to the KernelLauncher.