From f2654b634a6fd26e8677129d027d38adf10d5325 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9D=B4=ED=95=9C=EC=A2=85/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 22 Nov 2019 10:07:37 +0900 Subject: [PATCH] [neurun] Do not use name to check Operation type (#9110) Do not use name to check Operation type, use OpCode instead Signed-off-by: Hanjoung Lee --- runtime/neurun/core/src/graph/Graph.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/runtime/neurun/core/src/graph/Graph.cc b/runtime/neurun/core/src/graph/Graph.cc index aed9140..585e8fe 100644 --- a/runtime/neurun/core/src/graph/Graph.cc +++ b/runtime/neurun/core/src/graph/Graph.cc @@ -286,7 +286,7 @@ void Graph::makeSubgraphs( lower_info->addDefPermuteFactor(operand::PermuteFactor{backend, backend_layout}); } - if (node.name() == "Split") + if (node.opcode() == model::OpCode::Split) { // Ideally this condition must be like 'node.getOutputs().size() > 1' but // this is true for HashtableLookup also. TODO: Come up with more clever solution -- 2.7.4