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