diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-30 09:22:00 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-30 09:22:00 +0100 |
| commit | 393690387f2bab58c6638cfad37fcdc6c62d4cfc (patch) | |
| tree | 2e29b69e7a59d628f7ffa5f52770a33099e03743 /candle-core/src/cuda_backend.rs | |
| parent | 9b251133931cd450d4f7db734a06d6b113ddeb4f (diff) | |
| download | candle-393690387f2bab58c6638cfad37fcdc6c62d4cfc.tar.gz candle-393690387f2bab58c6638cfad37fcdc6c62d4cfc.tar.bz2 candle-393690387f2bab58c6638cfad37fcdc6c62d4cfc.zip | |
Support dilation in conv-transpose2d. (#671)
Diffstat (limited to 'candle-core/src/cuda_backend.rs')
| -rw-r--r-- | candle-core/src/cuda_backend.rs | 6 |
1 files changed, 0 insertions, 6 deletions
diff --git a/candle-core/src/cuda_backend.rs b/candle-core/src/cuda_backend.rs index 14a77b52..663f2319 100644 --- a/candle-core/src/cuda_backend.rs +++ b/candle-core/src/cuda_backend.rs @@ -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()..); |