Commit f52346cb authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[XLA] Rename {SqrtF32, SquareF32, ReciprocalF32} to {Sqrt, Square, Reciprocal}... 

[XLA] Rename {SqrtF32, SquareF32, ReciprocalF32} to {Sqrt, Square, Reciprocal} and move them to a new client library xla/client/lib/math.h. Remove the F32 type constraint.
Add an xla::Rqsrt function.

Move {Erf, Erfc, ErfInv, EvaluatePolynomial} to the same library.

[TF:XLA] Update many places in the bridge to use the new functions. Rewrite many of the training ops in operator notation.

PiperOrigin-RevId: 202948474
parent e0ebc3dc
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment