diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-01-06 17:40:30 +0000 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-01-18 14:03:46 +0000 |
| commit | b5a450a1acc1149f99f7bb06b10694fba554f4e3 (patch) | |
| tree | ac6ddf556c1656fd697a53b6ad5e961938f7d85d /src/core/NEON/kernels/NESoftmaxLayerKernel.h | |
| parent | d2447bb039c268aa21a5ca358cc2d91abe4f4d21 (diff) | |
| download | ComputeLibrary-b5a450a1acc1149f99f7bb06b10694fba554f4e3.tar.gz | |
Add SVE for Softmax
Implements COMPMID-3875
Change-Id: I38991eed3f4966db125862af066bfedff5994a25
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4854
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/NESoftmaxLayerKernel.h')
| -rw-r--r-- | src/core/NEON/kernels/NESoftmaxLayerKernel.h | 22 |
1 files changed, 7 insertions, 15 deletions
diff --git a/src/core/NEON/kernels/NESoftmaxLayerKernel.h b/src/core/NEON/kernels/NESoftmaxLayerKernel.h index adc2e57258..70e2417fc2 100644 --- a/src/core/NEON/kernels/NESoftmaxLayerKernel.h +++ b/src/core/NEON/kernels/NESoftmaxLayerKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2020 Arm Limited. + * Copyright (c) 2017-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -71,11 +71,7 @@ public: BorderSize border_size() const override; private: - using Logits1DMaxFunction = void(const ITensor &in, ITensor &out, const Window &window); - -private: - Logits1DMaxFunction *_func; - BorderSize _border_size; + BorderSize _border_size; }; /** Interface for softmax computation for QASYMM8 with pre-computed max. */ @@ -135,15 +131,11 @@ public: void run(const Window &window, const ThreadInfo &info) override; private: - using LogitsSoftmaxFunction = void(const ITensor &in, const ITensor &max, void *const tmp, ITensor &out, const float beta, - const Window &window); - - LogitsSoftmaxFunction *_func; - const ITensor *_input; - const ITensor *_max; - ITensor *_output; - float _beta; - ITensor *_tmp; //Temporary. Used internally + const ITensor *_input; + const ITensor *_max; + ITensor *_output; + float _beta; + ITensor *_tmp; //Temporary. Used internally }; } // namespace arm_compute #endif /*ARM_COMPUTE_NESOFTMAXLAYERKERNEL_H */ |