[Training] Add bf16 support to GatherElementsGrad.#20796

Merged

AdamLouly merged 4 commits intomainfrom adamlouly/bf16_gather_elements_grad

May 24, 2024

+56-6

Commits on May 23, 2024

initial bf16 support for gather elements grad op
Adam Louly
committed

Commits on May 24, 2024

redefine getElementType add stride tests
Adam Louly
committed
fix MLX tests, override func
Adam Louly
committed
resolve comments
Adam Louly
committed