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

[TF:XLA] Add support for the V2 variants of the FusedBatchNorm operators,... 

[TF:XLA] Add support for the V2 variants of the FusedBatchNorm operators, which support mixed precision training.

Until the necessary support for mixed precision fused batch norm is added to XLA, implement by casting to a common type.

PiperOrigin-RevId: 177468202
parent 93084701
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