[ML][single] Fix SingleShot.input attribute change

[platform/core/api/webapi-plugins.git] / src / ml / ml_instance.cc

diff --git a/src/ml/ml_instance.cc b/src/ml/ml_instance.cc
index 37e70e8..07b8a81 100644 (file)
--- a/src/ml/ml_instance.cc
+++ b/src/ml/ml_instance.cc
@@ -889,12 +889,15 @@ void MlInstance::MLSingleShotSetInputInfo(const picojson::value& args, picojson:
     return;
   }
 
+  TensorsInfo* clone = GetTensorsInfoManager().CloneTensorsInfo(in_tensors_info);
+
   auto ret = single_manager_.SetNativeInputInfo(id, in_tensors_info);
   if (!ret) {
     ReportError(ret, &out);
     return;
   }
 
+  out[kId] = picojson::value(static_cast<double>(clone->Id()));
   ReportSuccess(out);
 }