mlx/mlx/backend
Jagrit Digani 6686e61ca4
Reduce update (#783)
* Split reduction files to reduce compile times

* Add small and medium axis size specializations for row reductions

* Add non-row-reduction options for small and med kernels
2024-03-04 19:09:51 -08:00
..
accelerate Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00
common Convolution update (#651) 2024-02-28 20:11:16 -08:00
metal Reduce update (#783) 2024-03-04 19:09:51 -08:00
no_metal bindings for memory info (#761) 2024-03-01 19:51:58 -08:00