From 8a25b478cba0a9f1a339472b1a612f32bfedf269 Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EC=98=A4=ED=98=95=EC=84=9D/=EB=8F=99=EC=9E=91=EC=A0=9C?=
 =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?=
 =?utf8?q?=EC=A0=84=EC=9E=90?= <hseok82.oh@samsung.com>
Date: Thu, 6 Sep 2018 13:17:09 +0900
Subject: [PATCH] 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>
---
 runtimes/pure_arm_compute/src/model.cc | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

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;
 }
-- 
2.7.4