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/CPP/CPPScheduler.cpp

diff --git a/src/runtime/CPP/CPPScheduler.cpp b/src/runtime/CPP/CPPScheduler.cpp
index 41e1a2d..af6d8d7 100644
--- a/src/runtime/CPP/CPPScheduler.cpp
+++ b/src/runtime/CPP/CPPScheduler.cpp
@@ -363,7 +363,7 @@
 }
 #endif /* DOXYGEN_SKIP_THIS */
 
-void CPPScheduler::schedule_common(ICPPKernel *kernel, const Hints &hints, const std::vector<InputTensor> &inputs, const std::vector<OutputTensor> &outputs)
+void CPPScheduler::schedule_common(ICPPKernel *kernel, const Hints &hints, const InputTensorMap &inputs, const OutputTensorMap &outputs)
 {
     ARM_COMPUTE_ERROR_ON_MSG(!kernel, "The child class didn't set the kernel");
 
@@ -473,15 +473,15 @@
     }
 }
 
-void CPPScheduler::schedule_op(ICPPKernel *kernel, const Hints &hints, const std::vector<InputTensor> &inputs, const std::vector<OutputTensor> &outputs)
+void CPPScheduler::schedule_op(ICPPKernel *kernel, const Hints &hints, const InputTensorMap &inputs, const OutputTensorMap &outputs)
 {
     schedule_common(kernel, hints, inputs, outputs);
 }
 
 void CPPScheduler::schedule(ICPPKernel *kernel, const Hints &hints)
 {
-    const std::vector<InputTensor> inputs;
-    std::vector<OutputTensor>      outputs;
+    const InputTensorMap inputs;
+    OutputTensorMap      outputs;
     schedule_common(kernel, hints, inputs, outputs);
 }
 } // namespace arm_compute