huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 19:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bugfix for conv2d. (#343)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-08 16:20:00 +02:00
committed by GitHub
parent 89d3926c9b
commit 13ce68ff9b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1059,7 +1059,7 @@ impl<'a> Map2 for Conv2D<'a> {
let dst_idx = dst_idx + dst_c_idx * out_h * out_w; let dst_idx = dst_idx + dst_c_idx * out_h * out_w;
for dst_h in 0..out_h { for dst_h in 0..out_h {
let dst_idx = dst_idx + dst_h * out_w; let dst_idx = dst_idx + dst_h * out_w;
for dst_w in 0..out_h { for dst_w in 0..out_w {
let dst_idx = dst_idx + dst_w; let dst_idx = dst_idx + dst_w;
let mut d = T::zero(); let mut d = T::zero();
for offset_h in 0..p.k_h { for offset_h in 0..p.k_h {