huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
84ee870efda4fbbadafa67ee8615c76a8947ad24
candle/candle-nn/examples
History
Laurent Mazare 98d1242b8f im2col based conv2d (#802) 
* im2col implementation for conv2d.

* Fix for the im2col implementation to match the current conv2d.

* Small optimization.

* Add a cuda kernel.

* Handle arbitrary layouts.

* Im2Col cuda code.
2023-09-10 21:02:42 +01:00
..
basic_optimizer.rs
Add the optimizer trait. (#702)
2023-09-01 12:55:39 +01:00
cpu_benchmarks.rs
im2col based conv2d (#802)
2023-09-10 21:02:42 +01:00