Skip to main content
Log in

Mojo function

conv_cudnn

conv_cudnn[input_dim: DimList, filter_dim: DimList, output_dim: DimList, input_type: DType, filter_type: DType, output_type: DType](input: UnsafePointer[SIMD[input_type, 1]], filter: UnsafePointer[SIMD[filter_type, 1]], output: UnsafePointer[SIMD[output_type, 1]], stride: Index[2], dilation: Index[2], padding: Index[2], num_groups: Int, ctx: DeviceContext)