mlx/python
Jagrit Digani 776c3d226d
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>
2024-02-28 20:11:16 -08:00
..
mlx Add back compiled function signatures and docstrings (#749) 2024-02-27 13:18:59 -08:00
src Convolution update (#651) 2024-02-28 20:11:16 -08:00
tests Convolution update (#651) 2024-02-28 20:11:16 -08:00