[nnkit-mocotf] Set shape information in ModelSignature (#8289)

This commit will enable setting shape information in `ModelSignature`
at `nnkit-mocotf`

Signed-off-by: Seok NamKoong <sk.namkoong@samsung.com>

diff --git a/compiler/nnkit-mocotf/support/src/Backend.cpp b/compiler/nnkit-mocotf/support/src/Backend.cpp
index 51e1f7f..82d93bd 100644 (file)
--- a/compiler/nnkit-mocotf/support/src/Backend.cpp
+++ b/compiler/nnkit-mocotf/support/src/Backend.cpp
@@ -54,11 +54,13 @@ Backend::Backend(const char *pb_path, const char *info_path)
     if (parsed_tensor->kind() == ParsedTensor::Kind::Input)
     {
       sig.add_input(::moco::tf::TensorName(parsed_tensor->name()));
+      sig.shape(parsed_tensor->name(), parsed_tensor->shape());
       _inputs.emplace_back(std::move(parsed_tensor));
     }
     else
     {
       sig.add_output(::moco::tf::TensorName(parsed_tensor->name()));
+      sig.shape(parsed_tensor->name(), parsed_tensor->shape());
       _outputs.emplace_back(std::move(parsed_tensor));
     }
   }