mlx/python
Jacket 490c0c4fdc
[Fix] expand axes for dimension with integer indices in mlx_slice_update (#1035)
* Not sure if this is correct

* Format

* Edit tests

* Add negative test

* Format

* add one more test

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-04-29 07:57:28 -07:00
..
mlx avgpool, not maxpool (#1002) 2024-04-17 08:26:22 -07:00
src [Fix] expand axes for dimension with integer indices in mlx_slice_update (#1035) 2024-04-29 07:57:28 -07:00
tests [Fix] expand axes for dimension with integer indices in mlx_slice_update (#1035) 2024-04-29 07:57:28 -07:00