[moco-tf] Type inference for Conv2DBackpropInput (#7573)

This commit adds type inference for Conv2DBackpropInput.

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>

diff --git a/compiler/moco-tf/src/Dialect/TFTypeInferenceRule.cpp b/compiler/moco-tf/src/Dialect/TFTypeInferenceRule.cpp
index 8525768..5a513ea 100644 (file)
--- a/compiler/moco-tf/src/Dialect/TFTypeInferenceRule.cpp
+++ b/compiler/moco-tf/src/Dialect/TFTypeInferenceRule.cpp
@@ -39,6 +39,10 @@ struct TypeForwardAlgorithm final : public moco::tf::TFNodeVisitor<loco::DataTyp
   loco::DataType visit(const TFConst *node) { return node->dtype(); }
 
   loco::DataType visit(const TFConv2D *node) { return dtype_get(node->input()); }
+  loco::DataType visit(const TFConv2DBackpropInput *node)
+  {
+    return dtype_get(node->out_backprop());
+  }
   loco::DataType visit(const TFDepthwiseConv2dNative *node) { return dtype_get(node->input()); }
   loco::DataType visit(const TFFusedBatchNorm *node) { return dtype_get(node->input()); }
   loco::DataType visit(const TFIdentity *node) { return dtype_get(node->input()); }