commit ce0c67559cf03965acc8f212263a9f53205a0a3f
author Michalis Spyrou <michalis.spyrou@arm.com> Thu Jun 18 10:14:57 2020 +0100
committer Michalis Spyrou <michalis.spyrou@arm.com> Fri Jun 19 14:35:22 2020 +0000
tree c37105c72538108c46e5964cf49d15acd2d85980
parent 3be0b8c8d4e90bd264e9575dc2b6994ce8e14d50

COMPMID-3377: Async support to NEElementwiseUnaryLayerKernel kernels/functions

Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Change-Id: I208287b44ece051e95f891d43a691cb0ac6e56c5
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3419
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>

src/runtime/NEON/functions/NEReshapeLayer.cpp

diff --git a/src/runtime/NEON/functions/NEReshapeLayer.cpp b/src/runtime/NEON/functions/NEReshapeLayer.cpp
index daf358e..2b866b5 100644
--- a/src/runtime/NEON/functions/NEReshapeLayer.cpp
+++ b/src/runtime/NEON/functions/NEReshapeLayer.cpp
@@ -89,8 +89,9 @@
 
 void NEReshapeLayer::run()
 {
-    const InputTensor src{ TensorType::ACL_SRC, _impl->src };
-    OutputTensor      dst{ TensorType::ACL_DST, _impl->dst };
-    _impl->op->run({ src }, { dst }, {});
+    const InputTensorMap  src{ { TensorType::ACL_SRC, _impl->src } };
+    const OutputTensorMap dst{ { TensorType::ACL_DST, _impl->dst } };
+
+    _impl->op->run(src, dst, {});
 }
 } // namespace arm_compute