From 3b581ee3f978a6d62c96d54ceece020ca27e1d09 Mon Sep 17 00:00:00 2001 From: "William S. Moses" Date: Mon, 25 Sep 2023 23:25:05 -0500 Subject: [PATCH] Also fix depthwise grad im2col --- src/impl/depthwiseconv_im2col.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/impl/depthwiseconv_im2col.jl b/src/impl/depthwiseconv_im2col.jl index 30edc16e1..a909dc059 100644 --- 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