Convolution update (#651)

* Init steel conv and update Conv primitive * Update slow CPU implementation to support flipping and input dilation winograd conv routing Co-authored-by: Awni Hannun <[email protected]>
2024-02-28 20:11:16 -08:00
parent f5f18b704f
commit 776c3d226d
27 changed files with 2830 additions and 906 deletions
@@ -35,6 +35,7 @@ Operations
   convolve
   conv1d
   conv2d
+   conv_general
   cos
   cosh
   dequantize