[onert] Use the shape signature in baseloader (#14531)

This commit use the shape signature in baseloader. if there is an unknown shape in the circle, maintain the unknown shape ONE-DCO-1.0-Signed-off-by: youngsik kim <[email protected]>
Samsung · Jan 13, 2025 · 8543108 · 8543108
1 parent 0f38541
commit 8543108
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/runtime/onert/core/src/loader/BaseLoader.h b/runtime/onert/core/src/loader/BaseLoader.h
@@ -364,7 +364,8 @@ ir::OperandIndex BaseLoader<LoaderDomain>::loadOperand(const Tensor *tensor, ir:
 {
   ir::Shape shape;
   // Shape
-  const auto *tensor_shape = tensor->shape();
+  const auto *tensor_shape =
+    tensor->shape_signature() ? tensor->shape_signature() : tensor->shape();
   if (tensor_shape != nullptr)
   {
     for (const auto &dim : *tensor_shape)