IR graph: Fix getInput/getOutput (#949)

add checking of map search results

Signed-off-by: Vitaliy Cherepanov <v.cherepanov@samsung.com>

diff --git a/contrib/nnc/libs/core/src/core/IR/model/graph/graph.cpp b/contrib/nnc/libs/core/src/core/IR/model/graph/graph.cpp
index a2f0f80..4167992 100644 (file)
--- a/contrib/nnc/libs/core/src/core/IR/model/graph/graph.cpp
+++ b/contrib/nnc/libs/core/src/core/IR/model/graph/graph.cpp
@@ -16,11 +16,19 @@ namespace IR
 namespace model {
 
 INode::Ref Graph::getInput(const std::string &name) {
-  return _inputs.find(name)->second;
+  auto it = _inputs.find(name);
+  if (it == _inputs.end())
+    return nullptr;
+  else
+    return it->second;
 }
 
 INode::Ref Graph::getOutput(const std::string &name) {
-  return _outputs.find(name)->second;
+  auto it = _outputs.find(name);
+  if (it == _outputs.end())
+    return nullptr;
+  else
+    return it->second;
 }
 
 void Graph::accept(Visitor *visitor) {