[loco] Introduce DepthwiseFilterEncode to CanonicalShapeInferenceRule (#6548)

This commit introduces DepthwiseFilterEncode to
CanonicalShapeInferenceRule.

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

diff --git a/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp b/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp
index 78017a0..f5ca943 100644 (file)
--- a/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp
+++ b/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp
@@ -237,7 +237,12 @@ public:
   }
 
   // TODO Support DepthwiseConv2D
-  // TODO Support DepthwiseFilterEncode
+  // CASE: DepthwiseFilterEncode
+  loco::NodeShape visit(const loco::DepthwiseFilterEncode *node) final
+  {
+    auto input_tensor_shape = loco::shape_get(node->input()).as<loco::TensorShape>();
+    return loco::NodeShape{node->encoder()->shape(input_tensor_shape)};
+  }
 
   // CASE: EltwiseAdd
   loco::NodeShape visit(const loco::EltwiseAdd *node) final