Imported Upstream version 1.8.0

[platform/core/ml/nnfw.git] / runtime / onert / core / src / compiler / Linear.cc

diff --git a/runtime/onert/core/src/compiler/Linear.cc b/runtime/onert/core/src/compiler/Linear.cc
index b9db2f3..493ca1e 100644 (file)
--- a/runtime/onert/core/src/compiler/Linear.cc
+++ b/runtime/onert/core/src/compiler/Linear.cc
@@ -96,7 +96,7 @@ void Linear::planTensors(const ir::LoweredGraph &lowered_graph,
     }
 
     uses_map[ind] = obj.getUses().size();
-    def_map[ind] = obj.getDef().size(); // should be 1 or 0
+    def_map[ind] = obj.getDef().valid() ? 1 : 0;
 
     bool is_const = obj.isConstant();
     if (is_const)