Files
mlx/benchmarks
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
..
2024-04-11 07:27:53 -07:00
2023-12-08 11:31:47 -08:00