Thomas Santerre 754fa1e813 Add support for max_pool2d for Metal backend (#1863) ...

* first pass at implementation of maxpool2d

* Add definitions for other dtypes

* add tests for other dtypes

* Cosmetic tweaks + re-enable maxpool2d tests for metal.

---------

Co-authored-by: Laurent <laurent.mazare@gmail.com>

2024-03-18 08:33:30 +01:00

..

conv_tests.rs

Optimize the cat operation on contiguous tensors (#1855)

2024-03-17 10:49:13 +01:00

custom_op_tests.rs

Move the test-utils bits to a shared place. (#619)

2023-08-27 09:42:22 +01:00

display_tests.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

fortran_tensor_3d.pth

Add support for loading Fortran contiguous tensors (#1672)

2024-02-07 21:49:59 +01:00

grad_tests.rs

Optimize the cat operation on contiguous tensors (#1855)

2024-03-17 10:49:13 +01:00

indexing_tests.rs

Add a basic implementation for slice-assign. (#1377)

2023-11-26 17:31:22 +00:00

layout_tests.rs

Adding the test scaffolding.

2023-11-20 14:38:35 +01:00

npy.py

Fix the npy read function and add some testing. (#1080)

2023-10-12 15:25:05 +02:00

pool_tests.rs

Add support for max_pool2d for Metal backend (#1863)

2024-03-18 08:33:30 +01:00

pth_tests.rs

Add support for loading Fortran contiguous tensors (#1672)

2024-02-07 21:49:59 +01:00

pth.py

Add support for loading Fortran contiguous tensors (#1672)

2024-02-07 21:49:59 +01:00

quantized_tests.rs

Add a cuda kernel for dequantizing q8_0. (#1804)

2024-03-05 09:50:37 +01:00

serialization_tests.rs

Fix the npy read function and add some testing. (#1080)

2023-10-12 15:25:05 +02:00

tensor_tests.rs

Optimize the cat operation on contiguous tensors (#1855)

2024-03-17 10:49:13 +01:00

test_with_key.pt

Enhance pickle to retrieve state_dict with a given key (#1671)

2024-02-06 21:17:33 +01:00

test.npy

Fix the npy read function and add some testing. (#1080)

2023-10-12 15:25:05 +02:00

test.npz

Fix the npy read function and add some testing. (#1080)

2023-10-12 15:25:05 +02:00

test.pt

Fix: pth files don't load on Windows (#1661)

2024-02-06 08:50:55 +01:00