From ae69900a8899873979a84efe423ccea3323a8b99 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princi?= =?UTF-8?q?pal=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Tue, 17 Sep 2019 18:26:48 +0900 Subject: [moco-tf] get_node_shape helper in copy_shapedata (#7528) This will introduce get_node_shape helper in copy_shapedata of FixShape Signed-off-by: Jonghyun Park Signed-off-by: SaeHie Park --- compiler/moco-tf/src/Transforms/FixShapeTransform.cpp | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp b/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp index 934f5793a..93570dbbc 100644 --- a/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp +++ b/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp @@ -294,6 +294,22 @@ bool copy_shapedata(const loco::Node *lhs, const loco::Node *rhs, loco::Node *ds if (shape_inference_done(dst)) return false; + auto get_node_shape = [](const loco::Node *node, loco::NodeShape &out) { + if (loco::shape_known(node)) + { + out = loco::shape_get(node); + } + else + { + if (!shape_inference_done(node)) + return false; + + out = as_node_shape(node->annot()); + } + + return true; + }; + loco::NodeShape lhs_shape; loco::NodeShape rhs_shape; -- cgit v1.2.3