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
..
cpp Update pre-commit hooks (#984) 2024-04-11 07:27:53 -07:00
numpy Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
python Add groups to 2-D convolutions (#1129) 2024-05-22 20:01:44 -07:00