zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-11 19:56:40 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
554 Commits 39 Branches 69 Tags 1.7 GiB
9f9cb7a2ef
Commit Graph

2 Commits

Author SHA1 Message Date
Rifur13
9401507336
Add groups to 2-D convolutions (#1129) 
* Added groups to 2-D convolutions. Only implemented for **some** specializations.

Also fixed 1D grouped convs with different kernel strides and added more tests.

* fix channels condition
 2024-05-22 20:01:44 -07:00
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