From beb044a2f5161e71c4e140df3817a92a4a41153b 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: Thu, 19 Sep 2019 10:31:10 +0900 Subject: [PATCH] [moco-tf] Reduce dup code for copy_shapedata (#7592) This will update copy_shapedata() to use get_node_shape() to reduce duplicate codes in FixShapeTransform Signed-off-by: SaeHie Park --- .../moco-tf/src/Transforms/FixShapeTransform.cpp | 27 ++++------------------ 1 file changed, 4 insertions(+), 23 deletions(-) diff --git a/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp b/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp index 1ec5e07..b26058e 100644 --- a/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp +++ b/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp @@ -313,29 +313,10 @@ bool copy_shapedata(const loco::Node *lhs, const loco::Node *rhs, loco::Node *ds loco::NodeShape lhs_shape; loco::NodeShape rhs_shape; - if (loco::shape_known(lhs)) - { - lhs_shape = loco::shape_get(lhs); - } - else - { - if (!shape_inference_done(lhs)) - return false; - - lhs_shape = as_node_shape(lhs->annot()); - } - - if (loco::shape_known(rhs)) - { - rhs_shape = loco::shape_get(rhs); - } - else - { - if (!shape_inference_done(rhs)) - return false; - - rhs_shape = as_node_shape(rhs->annot()); - } + if (!get_node_shape(lhs, lhs_shape)) + return false; + if (!get_node_shape(rhs, rhs_shape)) + return false; if (lhs_shape.domain() != loco::Domain::Tensor || rhs_shape.domain() != loco::Domain::Tensor) { -- 2.7.4