Skip to content
Snippets Groups Projects
Commit ae2ae5ea authored by Jean-Marc Valin's avatar Jean-Marc Valin
Browse files
parent c7ba313a
No related branches found
No related tags found
No related merge requests found
......@@ -155,7 +155,7 @@ static inline void vec_sigmoid(float *y, const float *x, int N)
X = _mm256_loadu_ps(&x[i]);
Y = exp8_approx(X);
/* Compute as 1-1/(1+e^x) to avoid >1 values caused by the reciprocal approximation. */
Y = _mm256_sub_ps(one, _mm256_mul_ps(one, _mm256_rcp_ps(_mm256_add_ps(Y, one))));
Y = _mm256_sub_ps(one, _mm256_rcp_ps(_mm256_add_ps(Y, one)));
_mm256_storeu_ps(&y[i], Y);
}
for (;i<N;i++)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment