mlx/python
Awni Hannun f76a49e555
ExpandDims primitive (#1687)
* add squeeze primitive

* simplify squeeze, use in gather

* fix

* fix

* fix

* fix

* fix no cpu

* use squeeze in matmul and friends

* expand dims primitive

* comment
2024-12-10 16:39:07 -08:00
..
mlx No reshapes in quantized embedding (#1682) 2024-12-09 18:57:38 -08:00
src ExpandDims primitive (#1687) 2024-12-10 16:39:07 -08:00
tests ExpandDims primitive (#1687) 2024-12-10 16:39:07 -08:00