From: 오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Date: Thu, 6 Sep 2018 04:17:09 +0000 (+0900)
Subject: Use ExternalData for zero length value (#2618)
X-Git-Tag: 0.2~70
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=8a25b478cba0a9f1a339472b1a612f32bfedf269;p=platform%2Fcore%2Fml%2Fnnfw.git

Use ExternalData for zero length value (#2618)

Use ExternalData for object with no value (nullptr buffer from setOperandValue)

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>
---

diff --git a/runtimes/pure_arm_compute/src/model.cc b/runtimes/pure_arm_compute/src/model.cc
index a92164a..e549468 100644
--- a/runtimes/pure_arm_compute/src/model.cc
+++ b/runtimes/pure_arm_compute/src/model.cc
@@ -112,9 +112,16 @@ int ANeuralNetworksModel_setOperandValue(ANeuralNetworksModel *model, int32_t in
   const internal::tflite::operand::Index ind{index};
   auto &obj = model->deref().operands().at(ind);
 
-  using internal::tflite::operand::CachedData;
-
-  obj.data<CachedData>(reinterpret_cast<const uint8_t *>(buffer), length);
+  if (buffer == nullptr)
+  {
+    using internal::tflite::operand::ExternalData;
+    obj.data<ExternalData>(reinterpret_cast<const uint8_t *>(buffer), length);
+  }
+  else
+  {
+    using internal::tflite::operand::CachedData;
+    obj.data<CachedData>(reinterpret_cast<const uint8_t *>(buffer), length);
+  }
 
   return ANEURALNETWORKS_NO_ERROR;
 }