Module conv 

Convolution kernels

Enums

ConvStrategy

The strategy to be used when launching a convolution kernel.

ConvTranspose2dStrategy

The strategy to be used when launching a conv_transpose kernel.

Functions

conv_data_backward

Performs an N-dimensional convolution backwards data pass with the given strategy

conv_forward

Performs an N-dimensional convolution with the given strategy

conv_forward_nhwc

Performs an N-dimensional convolution with the given strategy on NHWC inputs/outputs

conv_transpose2d

Performs a 2D convolution with the given strategy

conv_weight_backward

Performs an N-dimensional convolution backwards pass with regard to weight, with the given strategy