diff options
author | Moritz Pflanzer <moritz.pflanzer@arm.com> | 2017-07-24 15:05:12 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:16:42 +0100 |
commit | a36ccf1b5c66a5f25bdde1ce69b60e49ed67bb41 (patch) | |
tree | cce25c0ce3dae31f820d25c7789ce9dc6ed5e47d /src/core/CL/cl_kernels/activation_layer.cl | |
parent | 8fda1cb6f4142133fff045a6f9c18778757c316c (diff) | |
download | ComputeLibrary-a36ccf1b5c66a5f25bdde1ce69b60e49ed67bb41.tar.gz |
COMPMID-417: Fix CL F16 ActivationLayer
Change-Id: I5b703d59661f6b50e3de4b2313be381b8d27030a
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/81951
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/CL/cl_kernels/activation_layer.cl')
-rw-r--r-- | src/core/CL/cl_kernels/activation_layer.cl | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/CL/cl_kernels/activation_layer.cl b/src/core/CL/cl_kernels/activation_layer.cl index 9f958610d6..a1e7002c8b 100644 --- a/src/core/CL/cl_kernels/activation_layer.cl +++ b/src/core/CL/cl_kernels/activation_layer.cl @@ -59,7 +59,7 @@ // Logistic Activation inline TYPE logistic_op(TYPE x) { - return DIV_OP(CONST_ONE, ADD_OP(CONST_ONE, EXP_OP(-x))); + return DIV_OP((TYPE)CONST_ONE, ADD_OP((TYPE)CONST_ONE, EXP_OP(-x))); } // Hyperbolic Tangent Activation inline TYPE tanh_op(TYPE x) @@ -84,7 +84,7 @@ inline TYPE lrelu_op(TYPE x) // Soft RELU Activation inline TYPE srelu_op(TYPE x) { - return LOG_OP(ADD_OP(CONST_ONE, EXP_OP(x))); + return LOG_OP(ADD_OP((TYPE)CONST_ONE, EXP_OP(x))); } // Absolute Activation inline TYPE abs_op(TYPE x) |