Laurent Mazare a6bcdfb269 Custom ops with a single argument (#214) ...

* Add the CustomOp1 trait.

* Add an example of custom op.

* Polish the custom op example.

* Add some backward pass test for custom ops.

2023-07-21 15:18:05 +01:00

..

custom_op_tests.rs

Custom ops with a single argument (#214)

2023-07-21 15:18:05 +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

Test the index op + bugfix. (#148)

2023-07-12 15:42:36 +01:00

layout_tests.rs

Iteration over strided blocks (#175)

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

tensor_tests.rs

Add the index-select op. (#209)

2023-07-20 14:01:03 +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