Struct WeightLayoutLaunch 

pub struct WeightLayoutLaunch<'a, R: Runtime> {
    pub channels: <FastDivmod as LaunchArg>::RuntimeArg<'a, R>,
    pub shape_k: <u32 as LaunchArg>::RuntimeArg<'a, R>,
    pub shape_n: <u32 as LaunchArg>::RuntimeArg<'a, R>,
    pub params: ConvolutionParams,
    pub config: GlobalMemoryConfig,
    /* private fields */
}

Fields

channels: <FastDivmod as LaunchArg>::RuntimeArg<'a, R>

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

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

params: ConvolutionParams

config: GlobalMemoryConfig

Implementations

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

pub fn new( channels: <FastDivmod as LaunchArg>::RuntimeArg<'a, R>, shape_k: <u32 as LaunchArg>::RuntimeArg<'a, R>, shape_n: <u32 as LaunchArg>::RuntimeArg<'a, R>, params: ConvolutionParams, config: GlobalMemoryConfig, ) -> Self

New kernel

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

pub fn from_args( client: &ComputeClient<R::Server>, problem: &ConvolutionProblem, params: ConvolutionParams, config: GlobalMemoryConfig, ) -> Self

Trait Implementations

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

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

Register the information of an argument to the KernelLauncher.