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 <awni@apple.com>
2025-09-01 04:24:36 +08:00 · 2024-02-28 20:11:16 -08:00
parent f5f18b704f
commit 776c3d226d
27 changed files with 2830 additions and 906 deletions
--- a/docs/src/python/ops.rst
+++ b/docs/src/python/ops.rst
@@ -35,6 +35,7 @@ Operations
   convolve
   conv1d
   conv2d
+   conv_general
   cos
   cosh
   dequantize