diff options
-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) |