From 352f75ef2b6fc33bf6c1c2fbaa8067f3967cce63 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Tue, 10 Sep 2019 10:47:25 +0900 Subject: [PATCH] [moco-tf] Use shape_inference_done for fix_shape COpCall (#7325) This will update fix_shape for COpCall to use shape_inference_done helper function Signed-off-by: SaeHie Park --- compiler/moco-tf/src/Transforms/FixShapeTransform.cpp | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp b/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp index 9364391..a03a32d 100644 --- a/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp +++ b/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp @@ -1211,9 +1211,7 @@ bool fix_shape(moco::tf::TFTanh *node) bool fix_shape(locoex::COpCall *node) { - // if node already has ShapeInferenceData, skip - auto shapedata = node->annot(); - if (shapedata != nullptr) + if (shape_inference_done(node)) return false; auto shape_data = make_shape_inference_data(node); -- 2.7.4