From 39f69fabfdb093fb498c9f966b5853bd260dbb66 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=B2=9C=EA=B5=90/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Wed, 14 Aug 2019 14:39:02 +0900 Subject: [PATCH] [loco/Service] FixedReshape type inference (#6595) This commit introduces type inference of FixedReshape Signed-off-by: Cheongyo Bahk --- compiler/loco/src/Service/TypeInference.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/compiler/loco/src/Service/TypeInference.cpp b/compiler/loco/src/Service/TypeInference.cpp index 2776fda..14bfcc8 100644 --- a/compiler/loco/src/Service/TypeInference.cpp +++ b/compiler/loco/src/Service/TypeInference.cpp @@ -113,7 +113,7 @@ struct CanonicalTypeForwardAlgorithm final : public loco::CanonicalNodeVisitorinput()); } 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(); } -- 2.7.4