From 1388d4789b91f06870ea69b86cefea89f14ab465 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: Tue, 20 Aug 2019 16:38:12 +0900 Subject: [PATCH] [moco-tf] Register shape inference pass (#6726) This commit registers shape inference pass to optimize stage transforms list. Signed-off-by: Cheongyo Bahk --- compiler/moco-tf/src/Transforms.h | 1 + 1 file changed, 1 insertion(+) diff --git a/compiler/moco-tf/src/Transforms.h b/compiler/moco-tf/src/Transforms.h index 22913c1..b1b9e84 100644 --- a/compiler/moco-tf/src/Transforms.h +++ b/compiler/moco-tf/src/Transforms.h @@ -30,5 +30,6 @@ #include "Transforms/ResolveRedundantReshape.h" #include "Transforms/ResolveReshapeWildcardDim.h" #include "Transforms/SimplifyDomainConversionTransform.h" +#include "Transforms/ShapeInferencePass.h" #endif // __MOCO_TF_TRANSFORMS_H__ -- 2.7.4