From c0fcda292d4bf138da227d205775a2aec62f0255 Mon Sep 17 00:00:00 2001 From: Stephen Johnson Date: Sat, 6 Jul 2019 18:55:13 -0700 Subject: [PATCH] The conv2D and conv2DBackpropFilter tuple values need to be switched. --- Sources/TensorFlow/Operators/NN.swift | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/Sources/TensorFlow/Operators/NN.swift b/Sources/TensorFlow/Operators/NN.swift index dc43087d3..ff57d7b53 100644 --- a/Sources/TensorFlow/Operators/NN.swift +++ b/Sources/TensorFlow/Operators/NN.swift @@ -185,9 +185,9 @@ func _vjpConv2DBackpropInput( let value = conv2DBackpropInput(x, shape: shape, filter: filter, strides: strides, padding: padding, dilations: dilations) return (value, { v in - (conv2DBackpropFilter(x, input: v, filterSizes: filter.shapeTensor, strides: strides, - padding: padding, dilations: dilations), - conv2D(v, filter: filter, strides: strides, padding: padding, dilations: dilations)) + (conv2D(v, filter: filter, strides: strides, padding: padding, dilations: dilations), + conv2DBackpropFilter(x, input: v, filterSizes: filter.shapeTensor, strides: strides, + padding: padding, dilations: dilations)) }) }