Lines Matching defs:sqrt

30 enum class AbsFn { abs, sqrt, rsqrt };
58   if (fn == AbsFn::sqrt) {
61     // sqrt(sqrt(a*b)) would avoid the pow, but will overflow more easily.
62     Value sqrt = b.create<math::SqrtOp>(max, fmfWithNaNInf);
64     result = b.create<arith::MulFOp>(sqrt, p025, fmfWithNaNInf);
66     Value sqrt = b.create<math::SqrtOp>(ratioSqPlusOne, fmfWithNaNInf);
67     result = b.create<arith::MulFOp>(max, sqrt, fmfWithNaNInf);
93 // atan2(y,x) = -i * log((x + i * y)/sqrt(x**2+y**2))
791     Value absSqrt = computeAbs(real, imag, fmf, b, AbsFn::sqrt);
796     // sin(atan2(0, inf)) = 0, sqrt(abs(inf)) = inf, but we can't multiply