Also fix depthwise

FluxML · Sep 26, 2023 · cd59223 · cd59223
1 parent 38247ae
commit cd59223
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/impl/depthwiseconv_im2col.jl b/src/impl/depthwiseconv_im2col.jl
@@ -131,7 +131,7 @@ function ∇depthwiseconv_data_im2col!(
                         gemm!(Val(false), Val(true), M, N, K, alpha, dy_ptr, w_ptr, T(0), col_ptr)
                     end
                 end
-                col2im!(view(dx, :, :, :, :, batch_idx), col_slice, cdims)
+                col2im!(view(dx, :, :, :, :, batch_idx), col_slice, cdims, beta)
             end
         end
     end