From c9458403982c11c0bbc6c36d7331c4084bd69156 Mon Sep 17 00:00:00 2001 From: PAWAN SASANKA AMMANAMANCHI Date: Sun, 28 Jul 2019 11:56:33 +0530 Subject: [PATCH] Fix in vjpConv2DBackpropFilter --- 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 2e30f521f..ae03bf15a 100644 --- a/Sources/TensorFlow/Operators/NN.swift +++ b/Sources/TensorFlow/Operators/NN.swift @@ -189,9 +189,9 @@ func _vjpConv2DBackpropFilter( let value = conv2DBackpropFilter(x, input: input, filterSizes: filterSizes, strides: strides, padding: padding, dilations: dilations) return (value, { v in - (conv2DBackpropInput(x, shape: filterSizes, filter: v, strides: strides, - padding: padding, dilations: dilations), - conv2D(input, filter: v, strides: strides, padding: padding, dilations: dilations)) + (conv2D(input, filter: v, strides: strides, padding: padding, dilations: dilations), + conv2DBackpropInput(x, shape: x.shapeTensor, filter: v, strides: strides, + padding: padding, dilations: dilations)) }) }