zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
Files
e4534dac1742ab29114764e1f779d55c6a90392a
mlx/tests/ops_tests.cpp
Awni Hannun 195b429d99 Put along axis + fixe for partition grad (#1430) 
* put along axis, fixes for partition grad

* zeros for arg reduce
2024-09-23 10:03:38 -07:00

109 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink