[loco] support DepthwiseConv2D of TypeForwardAlgorithm (#6404)

This commit is to support DepthwiseConv2D and DepthwiseFilterEncode in
CanonicalTypeForwardAlgorithm.

Signed-off-by: seongwoo <sw4670.chae@samsung.com>

diff --git a/compiler/loco/src/Service/TypeInference.cpp b/compiler/loco/src/Service/TypeInference.cpp
index 3c18cb3..40d77b2 100644 (file)
--- a/compiler/loco/src/Service/TypeInference.cpp
+++ b/compiler/loco/src/Service/TypeInference.cpp
@@ -98,8 +98,11 @@ struct CanonicalTypeForwardAlgorithm final : public loco::CanonicalNodeVisitor<l
   loco::DataType visit(const loco::BiasEncode *node) { return loco::dtype_get(node->input()); }
   loco::DataType visit(const loco::ConstGen *node) { return node->dtype(); }
   loco::DataType visit(const loco::Conv2D *node) { return loco::dtype_get(node->ifm()); }
-  // TODO Support DepthwiseConv2D
-  // TODO Support DepthwiseFilterEncode
+  loco::DataType visit(const loco::DepthwiseConv2D *node) { return loco::dtype_get(node->ifm()); }
+  loco::DataType visit(const loco::DepthwiseFilterEncode *node)
+  {
+    return loco::dtype_get(node->input());
+  }
   loco::DataType visit(const loco::EltwiseAdd *node) { return loco::dtype_get(node->lhs()); }
   loco::DataType visit(const loco::EltwiseDiv *node) { return loco::dtype_get(node->lhs()); }
   loco::DataType visit(const loco::EltwiseMul *node) { return loco::dtype_get(node->lhs()); }