Add sourcing tensor more than 5 dimensions (#2529)
author장지섭/동작제어Lab(SR)/Engineer/삼성전자 <jiseob.jang@samsung.com>
Fri, 31 Aug 2018 06:54:50 +0000 (15:54 +0900)
committer이춘석/동작제어Lab(SR)/Staff Engineer/삼성전자 <chunseok.lee@samsung.com>
Fri, 31 Aug 2018 06:54:50 +0000 (15:54 +0900)
This commit adds sourcing tensor more than 5 dimensions.

Signed-off-by: jiseob.jang <jiseob.jang@samsung.com>
runtimes/pure_arm_compute/src/execution.cc

index b1c287b..4100969 100644 (file)
@@ -362,7 +362,10 @@ int ANeuralNetworksExecution_setInput(ANeuralNetworksExecution *execution, int32
   }
   else
   {
-    throw std::runtime_error{"Not supported, yet"};
+    // NOTE TensorSource is much slower than specialized Source(s)
+    const auto &operand_shape = shape.asTensor();
+
+    asTensorSource(execution, input_type, index, operand_shape, buffer, length);
   }
 
   return ANEURALNETWORKS_NO_ERROR;