Lucas Newman 4a64d4bff1 Add support for grouped 1D convolutions to the nn API (#1444) ...

* Fix the weight shape for grouped convolutions from the nn API.

* Add tests.

* Pre-commit formatting.

* Add input validation.

* Use integer division instead of casting.

* docs

* nit

---------

Co-authored-by: Awni Hannun <awni@apple.com>

2024-09-28 06:41:07 -07:00

..

mlx

Add support for grouped 1D convolutions to the nn API (#1444)

2024-09-28 06:41:07 -07:00

src

allow take to work with integer index (#1440)

2024-09-26 15:58:03 -07:00

tests

Add support for grouped 1D convolutions to the nn API (#1444)

2024-09-28 06:41:07 -07:00