[loco/Service] FixedReshape type inference (#6595)

This commit introduces type inference of FixedReshape

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

diff --git a/compiler/loco/src/Service/TypeInference.cpp b/compiler/loco/src/Service/TypeInference.cpp
index 2776fda..14bfcc8 100644 (file)
--- a/compiler/loco/src/Service/TypeInference.cpp
+++ b/compiler/loco/src/Service/TypeInference.cpp
@@ -113,7 +113,7 @@ struct CanonicalTypeForwardAlgorithm final : public loco::CanonicalNodeVisitor<l
   loco::DataType visit(const loco::FeatureDecode *node) { return loco::dtype_get(node->input()); }
   loco::DataType visit(const loco::FeatureEncode *node) { return loco::dtype_get(node->input()); }
   loco::DataType visit(const loco::FilterEncode *node) { return loco::dtype_get(node->input()); }
-  // TODO Support FixedReshape
+  loco::DataType visit(const loco::FixedReshape *node) { return loco::dtype_get(node->input()); }
   loco::DataType visit(const loco::MaxPool2D *node) { return loco::dtype_get(node->ifm()); }
   loco::DataType visit(const loco::Push *node) { return loco::dtype_get(node->from()); }
   loco::DataType visit(const loco::Pull *node) { return node->dtype(); }