mlx/python
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
..
mlx Some fixes in docs (#1141) 2024-05-20 11:51:47 -07:00
src list based indexing (#1150) 2024-05-22 15:52:05 -07:00
tests Add groups to 2-D convolutions (#1129) 2024-05-22 20:01:44 -07:00