diff options
Diffstat (limited to 'reference_model/src/ops/activation_funcs.cc')
-rw-r--r-- | reference_model/src/ops/activation_funcs.cc | 4 |
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]); |