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

Support dilation in conv-transpose2d. (#671)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-30 09:22:00 +01:00
committed by GitHub
parent 9b25113393
commit 393690387f
4 changed files with 40 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-core/src/cuda_backend.rs
View File

@ -1046,12 +1046,6 @@ impl<'a> Map2 for ConvTranspose2D<'a> {
// Kernel shape: (c_in_k, c_out, h_k, w_k)
// Input shape: (b_size, c_in, h_in, w_in)
let p = &self.0;
if p.dilation != 1 {
crate::bail!(
"dilation {} is not supported for conv-transpose2d",
p.dilation
)
}
let (out_w, out_h) = (p.out_w(), p.out_h());
let dst_el = p.c_out * out_w * out_h * p.b_size;
let inp = &inp.slice(inp_l.start_offset()..);