Laurent Mazare 844d45cde4 Bugfix for the metal elu kernel. (#2490) ...

* Bugfix for the metal elu kernel.

* Add a test.

2024-09-21 15:03:19 +02:00

..

conv_tests.rs

Fix for backprop in ConvTranspose2D with stride of 2 (#2337)

2024-07-17 19:22:23 +02:00

custom_op_tests.rs

Preliminary support for inplace ops. (#1921)

2024-03-23 14:16:19 +01:00

display_tests.rs

…

fortran_tensor_3d.pth

…

grad_tests.rs

…

indexing_tests.rs

…

layout_tests.rs

Avoid copying the data on squeeze and unsqueeze. (#1884)

2024-03-20 13:04:36 +01:00

matmul_tests.rs

Add a minimal test for the metal bf16 matmul. (#2381)

2024-08-01 11:22:46 +02:00

npy.py

…

pool_tests.rs

Add avg_pool2d metal implementation for the metal backend (#1869)

2024-03-18 18:50:14 +01:00

pth_tests.rs

…

pth.py

…

quantized_tests.rs

Add the cuda dequantize f16 kernels. (#2137)

2024-04-28 20:05:05 +02:00

serialization_tests.rs

Add SliceSafetensors. (#2179)

2024-05-11 13:15:42 +02:00

tensor_tests.rs

Bugfix for the metal elu kernel. (#2490)

2024-09-21 15:03:19 +02:00

test_with_key.pt

…

test.npy

…

test.npz

…

test.pt

…