Skip to main content

launch_direct

cubek_convolution::kernels::forward::direct

Function launch_direct 

Source 
pub fn launch_direct<const N: usize>(
    client: &Client,
    tensors: DirectTensors,
    args: ConvolutionArgs<N>,
    groups: usize,
    dtype: ElemType,
) -> Result<(), ConvSetupError>
Expand description

Runs a direct convolution over N spatial dimensions.