diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-11-19 11:56:51 +0000 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-11-22 12:41:32 +0000 |
| commit | 303f0dbebf631b3db00d9d64e71018abbbe9d4fe (patch) | |
| tree | 631e70c9a8141f1262752829a64b3e33c7f1ee93 /src/core/NEON/kernels/NESoftmaxLayerKernel.cpp | |
| parent | 9d3a831d4131f8a8b37f127f11d36848d33e8496 (diff) | |
| download | ComputeLibrary-303f0dbebf631b3db00d9d64e71018abbbe9d4fe.tar.gz | |
COMPMID-1718: Extend DepthConvert to support Cast
Change-Id: I6ee2c0b670727fc808fa636c53ddfaec3a0036c9
Diffstat (limited to 'src/core/NEON/kernels/NESoftmaxLayerKernel.cpp')
| -rw-r--r-- | src/core/NEON/kernels/NESoftmaxLayerKernel.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/NESoftmaxLayerKernel.cpp b/src/core/NEON/kernels/NESoftmaxLayerKernel.cpp index 0f416defab..e9417ece44 100644 --- a/src/core/NEON/kernels/NESoftmaxLayerKernel.cpp +++ b/src/core/NEON/kernels/NESoftmaxLayerKernel.cpp @@ -34,7 +34,7 @@ #include "arm_compute/core/Utils.h" #include "arm_compute/core/Validate.h" #include "arm_compute/core/Window.h" -#include "arm_compute/core/utils/misc/Utility.h" +#include "arm_compute/core/utils/misc/SaturateCast.h" #include <algorithm> #include <arm_neon.h> @@ -667,7 +667,7 @@ void logits_1d_softmax_qasymm8(const ITensor &in, const ITensor &max, void *cons /* Run remaining elements */ for(; i < input_width; ++i) { - out_ptr[i] = utility::saturate_cast<qasymm8_t>(tmp_ptr[i] * sum_inversed); + out_ptr[i] = utils::cast::saturate_cast<qasymm8_t>(tmp_ptr[i] * sum_inversed); } } }, |