Files
candle/candle-metal-kernels
Thomas Santerre 04a61a9c72 Add avg_pool2d metal implementation for the metal backend (#1869)
* implement metal avg pool 2d

* fixX

* add suggested precision workaround for the accumulator
2024-03-18 18:50:14 +01:00
..
2023-11-30 11:30:31 +01:00
2023-11-20 14:12:56 +01:00

candle-metal-kernels

This crate contains Metal kernels used from candle.