[neurun] Fix TensorBuilder to add subtensors to Plan::operands (#3614)

Fix TensorBuilder to add subtensors to Plan::operands
Related issue is #3611

Signed-off-by: Poshshoev Dilshodzhon <d.poshshoev@samsung.com>

diff --git a/runtimes/neurun/src/backend/acl_cl/TensorBuilder.cc b/runtimes/neurun/src/backend/acl_cl/TensorBuilder.cc
index d591a36..dc5a64a 100644 (file)
--- a/runtimes/neurun/src/backend/acl_cl/TensorBuilder.cc
+++ b/runtimes/neurun/src/backend/acl_cl/TensorBuilder.cc
@@ -178,7 +178,14 @@ std::shared_ptr<::arm_compute::ITensor> TensorBuilder::tensorAt(const graph::ope
 std::shared_ptr<backend::operand::IObject>
 TensorBuilder::wrapTensor(const graph::operand::Index &ind)
 {
-  return std::make_shared<operand::Object>(_tensors.at(ind));
+  if (_tensors.find(ind) != _tensors.end())
+  {
+    return std::make_shared<operand::Object>(_tensors.at(ind));
+  }
+  else
+  {
+    return std::make_shared<operand::Object>(_subtensors.at(ind));
+  }
 }
 
 void TensorBuilder::iterate(const IterateFunction &fn)
@@ -187,6 +194,10 @@ void TensorBuilder::iterate(const IterateFunction &fn)
   {
     fn(it.first);
   }
+  for (auto it : _subtensors)
+  {
+    fn(it.first);
+  }
 }
 
 std::shared_ptr<::arm_compute::ICLTensor>