diff options
Diffstat (limited to 'candle-core/src/backprop.rs')
| -rw-r--r-- | candle-core/src/backprop.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-core/src/backprop.rs b/candle-core/src/backprop.rs index 26d73ea1..35619015 100644 --- a/candle-core/src/backprop.rs +++ b/candle-core/src/backprop.rs @@ -250,6 +250,7 @@ impl Tensor { out_padding, *stride, *dilation, + /* groups */ 1, )?; let sum_grad = grads.or_insert(arg)?; *sum_grad = sum_grad.add(&grad_arg)?; |