pub fn launch<R: Runtime, const N_SPATIAL: usize>(
strategy: &Strategy,
client: &ComputeClient<R>,
out_grad: MatmulInputHandle<R>,
weights: MatmulInputHandle<R>,
in_grad: TensorHandle<R>,
args: ConvolutionArgs<N_SPATIAL>,
dtypes: MatmulElems,
) -> Result<(), ConvSetupError>