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

More testing for avg-pool2d. (#366)

Browse Source 
* More testing for avg-pool2d.

* Another fix.

* Add a max-pool test with non-divisible kernel sizes.
This commit is contained in:
Laurent Mazare
2023-08-09 17:12:23 +02:00
committed by GitHub
parent dece0b8a76
commit cd225bd3b1
2 changed files with 37 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-core/src/cpu_backend.rs
View File

@ -660,8 +660,8 @@ impl Map1 for AvgPool2D {
let mut sum = T::zero();
for m in 0..k_h {
for n in 0..k_w {
let m = k_h * h_idx + m;
let n = k_w * w_idx + n;
let m = s_h * h_idx + m;
let n = s_w * w_idx + n;
sum += src[src_index + m * stride_h + n * stride_w]
}
}