diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2020-05-21 15:02:36 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-06-16 11:42:09 +0000 |
commit | bcd2352d7fd99a2f6aab220fa0c3b3f3119a1a4c (patch) | |
tree | a3e1880071bca828b1c58be71805ccce4b205e53 /arm_compute/runtime/NEON/functions/NESoftmaxLayer.h | |
parent | eae658453199d67a41deccbeb78e55b8eea9e966 (diff) | |
download | ComputeLibrary-bcd2352d7fd99a2f6aab220fa0c3b3f3119a1a4c.tar.gz |
COMPMID-3391: Implement Async interfaces
Change-Id: I8168cea5056ff48a0253ebb8c88ea549a3ea69a2
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3335
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NESoftmaxLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NESoftmaxLayer.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h b/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h index c5c83d8b5a..51d981de44 100644 --- a/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h +++ b/arm_compute/runtime/NEON/functions/NESoftmaxLayer.h @@ -25,11 +25,11 @@ #define ARM_COMPUTE_NESOFTMAXLAYER_H #include "arm_compute/core/NEON/kernels/NEFillBorderKernel.h" -#include "arm_compute/core/NEON/kernels/NEFlattenLayerKernel.h" -#include "arm_compute/core/NEON/kernels/NEReshapeLayerKernel.h" #include "arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h" #include "arm_compute/runtime/IFunction.h" #include "arm_compute/runtime/MemoryGroup.h" +#include "arm_compute/runtime/NEON/functions/NEFlattenLayer.h" +#include "arm_compute/runtime/NEON/functions/NEReshapeLayer.h" #include "arm_compute/runtime/Tensor.h" namespace arm_compute @@ -119,9 +119,9 @@ private: MemoryGroup _memory_group; NELogits1DMaxKernel _max_kernel; NELogits1DSoftmaxKernel<IS_LOG> _softmax_kernel; - std::unique_ptr<INEKernel> _flat_or_reshape_kernel_ptr; + std::unique_ptr<IFunction> _flat_or_reshape_ptr; NEFillBorderKernel _fill_border_kernel; - NEReshapeLayerKernel _reshape_kernel; + NEReshapeLayer _reshape; Tensor _max; Tensor _tmp; Tensor _input_flattened; |