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