[luci/service] Replace shape_get to circle_shape function

This commit replaces shape_get to circle_shape function. ONE-DCO-1.0-Signed-off-by: Chansu Park <1265pcs@gmail.com>
Samsung · Sep 8, 2024 · 52a2f8a · 52a2f8a
1 parent ea8d0d9
commit 52a2f8a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/compiler/luci/service/src/Nodes/CircleConv2D.cpp b/compiler/luci/service/src/Nodes/CircleConv2D.cpp
@@ -52,8 +52,10 @@ loco::TensorShape Algorithm::visit(const luci::CircleConv2D *node)
 {
   LOGGER(l);
 
-  auto ifm_shape = luci::shape_get(node->input()).as<loco::TensorShape>();  // in NHWC
-  auto ker_shape = luci::shape_get(node->filter()).as<loco::TensorShape>(); // in OHWI
+  auto ifm = loco::must_cast<luci::CircleNode *>(node->input());
+  auto ifm_shape = circle_shape(ifm); // in NHWC
+  auto ker = loco::must_cast<luci::CircleNode *>(node->filter());
+  auto ker_shape = circle_shape(ker); // in OHWI
 
   assert(ifm_shape.rank() == 4);
   assert(ker_shape.rank() == 4);

diff --git a/compiler/luci/service/src/Nodes/CircleDepthwiseConv2D.cpp b/compiler/luci/service/src/Nodes/CircleDepthwiseConv2D.cpp
@@ -49,8 +49,11 @@ namespace sinf
 
 loco::TensorShape Algorithm::visit(const luci::CircleDepthwiseConv2D *node)
 {
-  auto ifm_shape = luci::shape_get(node->input()).as<loco::TensorShape>();  // in NHWC
-  auto ker_shape = luci::shape_get(node->filter()).as<loco::TensorShape>(); // in 1 H W CM
+  auto ifm = loco::must_cast<luci::CircleNode *>(node->input());
+  auto ifm_shape = circle_shape(ifm); // in NHWC
+  auto ker = loco::must_cast<luci::CircleNode *>(node->filter());
+  auto ker_shape = circle_shape(ker); // in 1 H W CM
+
 
   assert(ifm_shape.rank() == 4);
   assert(ker_shape.rank() == 4);