Laurent Mazare a094dc503d Add a cuda kernel for avg-pool2d. (#440) ...

* Add a cuda kernel for avg-pool2d.

* Avoid running out of bounds.

* Finish wiring the avg pool kernel + add some testing.

* Support for max-pool + testing.

2023-08-14 12:32:05 +01:00

..

conv_tests.rs

Add a naive conv2d cuda kernel. (#438)

2023-08-14 10:34:42 +01:00

custom_op_tests.rs

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

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

display_tests.rs

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

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

grad_tests.rs

More accelerate optimizations (#427)

2023-08-13 12:53:34 +01:00

indexing_tests.rs

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

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

layout_tests.rs

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

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

pool_tests.rs

Add a cuda kernel for avg-pool2d. (#440)

2023-08-14 12:32:05 +01:00

tensor_tests.rs

Further randn tweaks: use the appropriate rng rather than the f64 one, some cleanup. (#383)

2023-08-10 05:48:19 +01:00

test_utils.rs

More accelerate optimizations (#427)

2023-08-13 12:53:34 +01:00