diff options
| author | Anthony Barbier <anthony.barbier@arm.com> | 2018-08-10 17:36:36 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | 3a6163ed0c2d0ab4cac0456e8f66c704c6ad10c2 (patch) | |
| tree | f2e34b47fb0b8bf1fac18c337a67c0145def7b7b /src/core/NEON/kernels/NESoftmaxLayerKernel.cpp | |
| parent | e3f1bd10e71075f7d8b406698809666e1eb4e7ae (diff) | |
| download | ComputeLibrary-3a6163ed0c2d0ab4cac0456e8f66c704c6ad10c2.tar.gz | |
COMPMID-1499: Fixed issues to build for FP16 on Android
Change-Id: I7cd15e9115b5c6f544005528d69061751286be11
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/143708
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele DiGiorgio <michele.digiorgio@arm.com>
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 4041b623b1..484e58b79b 100644 --- a/src/core/NEON/kernels/NESoftmaxLayerKernel.cpp +++ b/src/core/NEON/kernels/NESoftmaxLayerKernel.cpp @@ -392,7 +392,7 @@ void logits_1d_max(const ITensor &in, ITensor &out, const Window &window) const auto out_ptr = reinterpret_cast<T *>(output.ptr()); // Init max value - auto vec_max = vdup_n<vec_16_byte_t<T>>(std::numeric_limits<T>::lowest()); + auto vec_max = vdup_n<vec_16_byte_t<T>>(support::cpp11::lowest<T>()); // Loop over input row for(const T *it = in_ptr; it < (in_ptr + input_width); it += vec_size_of(vec_max)) @@ -694,7 +694,7 @@ void logits_1d_softmax_float(const ITensor &in, const ITensor &max, void *const { auto vec_elements = vld<vec_16_byte_t<T>>(in_ptr + i); vec_elements = vsub(vec_elements, vec_max); - vec_elements = vexp(vmul_n(vec_elements, beta)); + vec_elements = vexp(vmul_n(vec_elements, static_cast<T>(beta))); vec_sum = vadd(vec_sum, vec_elements); vst(tmp_ptr + i, vec_elements); } |