Skip to content
This repository was archived by the owner on Jul 1, 2023. It is now read-only.

Commit 09c68d7

Browse files
sjaz24rxwei
authored andcommitted
Swap pullback return values in '_vjpConv2DBackpropInput'. (#333)
1 parent 8e1a71d commit 09c68d7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

Sources/TensorFlow/Operators/NN.swift

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -185,9 +185,9 @@ func _vjpConv2DBackpropInput<Scalar: TensorFlowFloatingPoint>(
185185
let value = conv2DBackpropInput(x, shape: shape, filter: filter,
186186
strides: strides, padding: padding, dilations: dilations)
187187
return (value, { v in
188-
(conv2DBackpropFilter(x, input: v, filterSizes: filter.shapeTensor, strides: strides,
189-
padding: padding, dilations: dilations),
190-
conv2D(v, filter: filter, strides: strides, padding: padding, dilations: dilations))
188+
(conv2D(v, filter: filter, strides: strides, padding: padding, dilations: dilations),
189+
conv2DBackpropFilter(x, input: v, filterSizes: filter.shapeTensor, strides: strides,
190+
padding: padding, dilations: dilations))
191191
})
192192
}
193193

0 commit comments

Comments
 (0)