Skip to main content

launch_ref

Function launch_ref 

Source
pub fn launch_ref<R: Runtime, const N_SPATIAL: usize>(
    strategy: &Strategy,
    client: &ComputeClient<R>,
    inputs: ConvolutionInputs<R>,
    args: ConvolutionArgs<N_SPATIAL>,
    dtypes: MatmulElems,
) -> Result<(), ConvSetupError>
Expand description

The single public convolution entry point.

Routes the inputs (whose discriminant is the operation) and strategy (algorithm + tile-matmul kind, optionally a forced blueprint) into the right generic Routine and per-operation launch helper.