diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-11-10 10:26:11 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | 4df76c9c56bebabe0eafb7471b83ad731d8ad46a (patch) | |
tree | 006b352c55e084dd598784742f45458f06573384 | |
parent | 7d6bf6aa901d5fe08f37276533ef0ba45f1f69ff (diff) | |
download | ComputeLibrary-4df76c9c56bebabe0eafb7471b83ad731d8ad46a.tar.gz |
COMPMID-661: Fix beta in softmax_layer kernel
Change-Id: Iebb01c9e3d87ed17fa7991e157cfd15d795a53f1
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/95281
Reviewed-by: Robert Hughes <robert.hughes@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
-rw-r--r-- | src/core/CL/cl_kernels/softmax_layer.cl | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/CL/cl_kernels/softmax_layer.cl b/src/core/CL/cl_kernels/softmax_layer.cl index 5bc43ef144..fc45ae6ff6 100644 --- a/src/core/CL/cl_kernels/softmax_layer.cl +++ b/src/core/CL/cl_kernels/softmax_layer.cl @@ -363,7 +363,7 @@ __kernel void softmax_layer_max_shift_exp_sum_serial( #ifdef BETA // Initialize beta VEC_DATA_TYPE(DATA_TYPE, VECTOR_SIZE) - beta = (VEC_DATA_TYPE(DATA_TYPE, VECTOR_SIZE))BETA_VAL; + beta = (VEC_DATA_TYPE(DATA_TYPE, VECTOR_SIZE))BETA; #endif /* BETA */ // Initialize local maximum |