diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-14 12:32:05 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-14 12:32:05 +0100 |
| commit | a094dc503d69a6ca3db71098ebc26d0d2f2a33a6 (patch) | |
| tree | fa1242f898b216db10453eb357c6169f761becaf /candle-wasm-examples | |
| parent | 34f4b3187ec0e30e439a89ff7211a74050bc2358 (diff) | |
| download | candle-a094dc503d69a6ca3db71098ebc26d0d2f2a33a6.tar.gz candle-a094dc503d69a6ca3db71098ebc26d0d2f2a33a6.tar.bz2 candle-a094dc503d69a6ca3db71098ebc26d0d2f2a33a6.zip | |
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.
Diffstat (limited to 'candle-wasm-examples')
0 files changed, 0 insertions, 0 deletions