mlx/tests
Diogo 449b43762e
Add inner / outer op (#348)
* inner / outer impl

* python tests

* ops list and ack

* updated descriptions

* use test helper

* removed dtype check and flatten outer to 1-D

* updated docs

* just use the reshape to flatten
2024-01-07 09:01:09 -08:00
..
allocator_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
arg_reduce_tests.cpp Spelling (#342) 2024-01-01 21:08:17 -08:00
array_tests.cpp Make array conform to the Python Buffer Protocol (#323) 2024-01-05 15:58:33 -08:00
autograd_tests.cpp Spelling (#342) 2024-01-01 21:08:17 -08:00
blas_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
CMakeLists.txt linalg.norm (#187) 2023-12-26 19:42:04 -08:00
creations_tests.cpp Spelling (#342) 2024-01-01 21:08:17 -08:00
device_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
eval_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
fft_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
graph_optimize_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
linalg_tests.cpp linalg.norm (#187) 2023-12-26 19:42:04 -08:00
load_tests.cpp Safetensor support (#215) 2023-12-27 02:06:55 -08:00
metal_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
ops_tests.cpp Add inner / outer op (#348) 2024-01-07 09:01:09 -08:00
random_tests.cpp random.uniform must respect dtype, even if lower precision than "low" (#280) 2023-12-24 07:04:43 -08:00
scheduler_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
utils_tests.cpp Added mx.stack c++ frontend impl (#123) 2023-12-14 13:21:19 -08:00
vmap_tests.cpp Add move and swap axis, and vmap for slice, concat, and gather (#158) 2023-12-14 12:59:12 -08:00