Laurent Mazare c950a5c6b1 Cuda support for the mnist training. (#277) ...

* Cuda support for the mnist training.

* min/max fix + testing.

* Add the argmin/argmax tests.

* More cuda support for argmin/argmax.

* Cuda kernels for argmin and argmax.

2023-07-29 19:48:04 +01:00

..

custom_op_tests.rs

Add binary and ternary custom ops. (#217)

2023-07-21 17:29:50 +01:00

display_tests.rs

Add some display tests + bugfixes.

2023-06-27 21:37:28 +01:00

grad_tests.rs

Add some epsilon tolerance to grad tests so that they work on cuda / mkl. (#213)

2023-07-21 12:45:14 +01:00

indexing_tests.rs

Do not panic on empty ranges. (#257)

2023-07-27 09:28:47 +01:00

layout_tests.rs

Iteration over strided blocks (#175)

2023-07-15 21:30:35 +01:00

tensor_tests.rs

Cuda support for the mnist training. (#277)

2023-07-29 19:48:04 +01:00

test_utils.rs

Add some epsilon tolerance to grad tests so that they work on cuda / mkl. (#213)

2023-07-21 12:45:14 +01:00