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