Skip to content

Commit f283696

Browse files
committed
grad filter fixup
1 parent 77254d8 commit f283696

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/nnpack/interface.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,5 +89,6 @@ end
8989
function ∇conv_filter!(dw::A, dy::A, x::A, w::A; pad = 0, stride = 1, dilation = 1, algo = UInt32(0), flipkernel = 0) where A<:AbstractArray{Float32, 4}
9090
flipkernel == 0 && (w = flipweight(w))
9191
nnp_convolution_kernel_gradient(dw, x, dy, w, padding = pad, stride = stride, algo = algo)
92-
flipkernel == 0 ? flipweight(dw) : dw
92+
flipkernel && (dw .= flipkernel(dw))
93+
return dw
9394
end

0 commit comments

Comments
 (0)