Skip to content

Commit c514d05

Browse files
authored
Merge pull request #115 from FluxML/ap/nnpack-patch
NNPACK Convolution Patch
2 parents ec79173 + ad55659 commit c514d05

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/nnpack/impl.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ end
1313
threadpool = select_threadpool(cdims, size(y, 4))
1414

1515
if flipkernel(cdims) == 0
16-
w .= flipweight(w)
16+
w = flipweight(w)
1717
end
1818

1919
nnp_convolution_output(y, x, w, b, algo = algo, padding = padding(cdims),
@@ -26,7 +26,7 @@ end
2626
threadpool = select_threadpool(cdims, size(y, 4))
2727

2828
if flipkernel(cdims) == 0
29-
w .= flipweight(w)
29+
w = flipweight(w)
3030
end
3131

3232
nnp_convolution_input_gradient(dx, dy, w, algo = algo, padding = padding(cdims),

0 commit comments

Comments
 (0)