@@ -164,18 +164,18 @@ if is_nnpack_available()
164
164
end
165
165
end
166
166
167
- function conv (x, w:: AbstractArray{T, N} ; stride = 1 , pad = 0 , dilation = 1 ) where {T, N}
167
+ function conv (x, w:: AbstractArray{T, N} ; stride = 1 , pad = 0 , dilation = 1 , flipped = false ) where {T, N}
168
168
stride = expand (Val (N- 2 ), stride)
169
169
pad = expand (Val (N- 2 ), pad)
170
170
dilation = expand (Val (N- 2 ), dilation)
171
- cdims = DenseConvDims (x, w; stride = stride, padding = pad, dilation = dilation)
171
+ cdims = DenseConvDims (x, w; stride = stride, padding = pad, dilation = dilation, flipkernel = flipped )
172
172
return conv (x, w, cdims)
173
173
end
174
174
175
- function depthwiseconv (x, w:: AbstractArray{T, N} ; stride = 1 , pad = 0 , dilation = 1 ) where {T, N}
175
+ function depthwiseconv (x, w:: AbstractArray{T, N} ; stride = 1 , pad = 0 , dilation = 1 , flipped = false ) where {T, N}
176
176
stride = expand (Val (N- 2 ), stride)
177
177
pad = expand (Val (N- 2 ), pad)
178
178
dilation = expand (Val (N- 2 ), dilation)
179
- cdims = DepthwiseConvDims (x, w; stride = stride, padding = pad, dilation = dilation)
179
+ cdims = DepthwiseConvDims (x, w; stride = stride, padding = pad, dilation = dilation, flipkernel = flipped )
180
180
return depthwiseconv (x, w, cdims)
181
181
end
0 commit comments