[moco-tf] Fix for ui_0004 seg-fault (#9185)

This will handle invalid info file error as oops::UserExn

Signed-off-by: SaeHie Park <saehie.park@samsung.com>

diff --git a/compiler/moco-tf/src/Frontend.cpp b/compiler/moco-tf/src/Frontend.cpp
index a446bff..8ffb588 100644 (file)
--- a/compiler/moco-tf/src/Frontend.cpp
+++ b/compiler/moco-tf/src/Frontend.cpp
@@ -136,6 +136,10 @@ void set_input_shape(const moco::ModelSignature &signature, tensorflow::GraphDef
 
     auto node_shape = n.mutable_attr()->at("shape").mutable_shape();
     auto sig_shape = signature.shape(n.name() + ":0");
+    if (sig_shape == nullptr)
+    {
+      throw oops::UserExn("Please check model info file. Node not found in model file", n.name());
+    }
     if (node_shape->unknown_rank() || !node_shape->dim_size())
     {
       node_shape->clear_unknown_rank();