Laurent Mazare cc76c63202 Use index-select for the embeddings as it supports backprop. (#298)

2023-08-01 20:44:43 +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

Use index-select for the embeddings as it supports backprop. (#298)

2023-08-01 20:44:43 +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