Add gradient for ResizeBilinear

Author: chrberen

tensorflow-ops/src/TensorFlow/Gradient.hs

@@ -817,6 +817,18 @@ opGrad "Tile" _ [toT -> x, toT -> multiples] [dz] =
     axes = CoreOps.range 0 (CoreOps.size splitShape) (2 :: Tensor Build Int32)
     reshapedDz = CoreOps.reshape dz splitShape
 
+opGrad "ResizeBilinear" nodeDef [toT -> x, _] [dz] =
+    [ Just $ CoreOps.cast $ CoreOps.resizeBilinear'
+               (opAttr "align_corners" .~ align)
+               dz
+               xSize32
+
+    , Nothing
+    ]
+  where
+    xSize32 = flatSlice (shape (x :: Tensor Build a)) 1 2
+    align = lookupAttr nodeDef "align_corners" :: Bool
+
 opGrad "ZerosLike" _ _ _ = [Nothing]
 opGrad "Fill" _ _ [dz] = [Nothing, Just $ sum dz rx]
   where
@@ -894,6 +906,7 @@ numOutputs o =
         "Sum" -> 1
         "Tanh" -> 1
         "Tile" -> 1
+        "ResizeBilinear" -> 1
         "Transpose" -> 1
         "TruncatedNormal" -> 1
         "VarHandleOp" -> 1

tensorflow-ops/tests/GradientTest.hs

@@ -32,7 +32,7 @@ import Control.Monad(forM_, replicateM, zipWithM)
 import Control.Monad.IO.Class (liftIO)
 
 import qualified TensorFlow.Core as TF
-import qualified TensorFlow.GenOps.Core as TF (conv2DBackpropInput', max, maximum, tile, pad, batchToSpaceND, spaceToBatchND, squeeze, sqrt, slice, shape)
+import qualified TensorFlow.GenOps.Core as TF (conv2DBackpropInput', max, maximum, resizeBilinear, tile, pad, batchToSpaceND, spaceToBatchND, squeeze, sqrt, slice, shape)
 import qualified TensorFlow.Gradient as TF
 import qualified TensorFlow.Ops as TF hiding (zeroInitializedVariable, shape)
 import qualified TensorFlow.Output as TF
@@ -429,6 +429,19 @@ testTile2DGrad = testCase "testTileGrad2D" $ do
     shapeX @=? (shapeDX :: V.Vector Int32)
     V.fromList [6, 6, 6, 6, 6, 6::Float] @=? (dx :: V.Vector Float)
 
+testResizeBilinearGrad :: Test
+testResizeBilinearGrad = testCase "testResizeBilinearGrad" $ do
+    (dx, shapeDX, shapeX) <- TF.runSession $ do
+        let shape = TF.vector [1, 3, 2, 1 :: Int32]
+        x <- TF.render $ TF.fill shape (TF.scalar (1::Float))
+        let outSize = TF.vector [6, 4 :: Int32]
+        let y = TF.resizeBilinear x outSize
+
+        [dx] <- TF.gradients y [x]
+        TF.run (dx, TF.shape dx, TF.shape x)
+    shapeX @=? (shapeDX :: V.Vector Int32)
+    V.fromList [1, 1, 1, 1, 1, 1::Float] @=? (dx :: V.Vector Float)
+
 matMulGradient :: Test
 matMulGradient = testCase "matMulGradients" $ do
 
@@ -553,6 +566,7 @@ main = defaultMain
             , testFillGrad
             , testTileGrad
             , testTile2DGrad
+            , testResizeBilinearGrad
             , matMulGradient
             , matMulGradGrad
             , matMulTransposeGradient (False, False)