mlx/tests
__mo_san__ 52e1589a52
implemented Flatten Module (#149)
* implemented flatten op

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-16 21:54:37 -08:00
..
allocator_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
arg_reduce_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
array_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
autograd_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
blas_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
CMakeLists.txt awni's commit files 2023-11-29 10:30:41 -08:00
creations_tests.cpp copyright + ack 2023-11-30 11:12:53 -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
load_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
metal_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
ops_tests.cpp implemented Flatten Module (#149) 2023-12-16 21:54:37 -08:00
random_tests.cpp random generation fix (#80) 2023-12-08 10:40:57 -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