[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 <jh1302.park@samsung.com>
Signed-off-by: SaeHie Park <saehie.park@samsung.com>

1 files changed, 16 insertions, 0 deletions

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<ShapeInferenceData>());
+    }
+
+    return true;
+  };
+
   loco::NodeShape lhs_shape;
   loco::NodeShape rhs_shape;