aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/ops/activation_funcs.cc
diff options
context:
space:
mode:
Diffstat (limited to 'reference_model/src/ops/activation_funcs.cc')
-rw-r--r--reference_model/src/ops/activation_funcs.cc4
1 files changed, 3 insertions, 1 deletions
diff --git a/reference_model/src/ops/activation_funcs.cc b/reference_model/src/ops/activation_funcs.cc
index 46234e2..09ef6f6 100644
--- a/reference_model/src/ops/activation_funcs.cc
+++ b/reference_model/src/ops/activation_funcs.cc
@@ -63,7 +63,9 @@ int OpSigmoid<Rank, Dtype>::register_fcn()
case DType_FP16:
case DType_BF16:
case DType_FP32:
- this->fcn = [](InEigenType a) -> OutEigenType { return fpTrunc<Dtype>(1.0 / (1.0 + (expf(-1.0 * a)))); };
+ this->fcn = [](InEigenType a) -> OutEigenType {
+ return fpTrunc<Dtype>(1.f / (1.f + (expf(-1.f * a))));
+ };
break;
default:
ERROR_IF(true, "unsupported DType %s", EnumNamesDType()[Dtype]);