Fix in vjpConv2DBackpropFilter (#397)

Author: Shashi456

Sources/TensorFlow/Operators/NN.swift

@@ -189,9 +189,9 @@ func _vjpConv2DBackpropFilter<Scalar: TensorFlowFloatingPoint>(
     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))
     })
 }