mlx/benchmarks
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
..
cpp Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00
numpy Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
python Convolution update (#651) 2024-02-28 20:11:16 -08:00