Change the way getting layout of node in PermutationInsertionPass (#6148)

This commit changes the way getting layout of node in PermuttationInsertionPass from model::Subgraph to operation::LowerInfo.

Signed-off-by: jiseob.jang <jiseob.jang@samsung.com>

diff --git a/runtimes/neurun/core/src/graph/pass/PermutationInsertionPass.cc b/runtimes/neurun/core/src/graph/pass/PermutationInsertionPass.cc
index 447857c..b27ab05 100644 (file)
--- a/runtimes/neurun/core/src/graph/pass/PermutationInsertionPass.cc
+++ b/runtimes/neurun/core/src/graph/pass/PermutationInsertionPass.cc
@@ -89,7 +89,7 @@ void PermutationInsertionPass::callback(const model::OperandIndex &index, model:
       auto subg_index = _graph.subgraphs().getOperation(use);
       auto subg_li = _graph.getLowerInfo(subg_index);
       assert(subg_li);
-      const auto subg_layout = _graph.subgraphs().at(subg_index).getLayout();
+      const auto subg_layout = subg_li->layout();
       const backend::Backend *backend = subg_li->backend();
       assert(backend);
       auto use_node_inputs = operation.getInputs();