Diogo dc2edc762c added tri / tril / triu (#170) ...

* added tri / tril / triu

* fixed tests

* ctest tests

* tri overload and simplified tests

* changes from comment

* more tests for m

* ensure assert if not 2-D

* remove broadcast_to

* minor tweaks

---------

Co-authored-by: Awni Hannun <awni@apple.com>

2023-12-15 17:30:34 -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

added tri / tril / triu (#170)

2023-12-15 17:30:34 -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