huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add some conv1d test + bugfix using padding. (#349)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-08 21:50:20 +02:00
committed by GitHub
parent 1e6dbeac01
commit 608b2358c6
3 changed files with 47 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1011,7 +1011,7 @@ impl<'a> Map2 for Conv1D<'a> {
let dst_idx = dst_idx + dst_l;
let mut d = T::zero();
for offset in 0..p.k_size {
let src_l_plus = p.stride * dst_l + offset;
let src_l_plus = p.stride * dst_l + offset + k_over_2 - p.padding;
// inp[bidx, src_c_idx, dst_l + offset - k//2] * k[dst_c_idx, src_c_idx, offset]
if k_over_2 <= src_l_plus && src_l_plus < k_over_2 + p.l_in {
let src_l = src_l_plus - k_over_2;