Commit 759690f0 authored by Reed Wanderman-Milne's avatar Reed Wanderman-Milne Committed by TensorFlower Gardener
Browse files

Add float16 support to tf.nn.fused_batch_norm on the GPU. 

Scale, offset, mean, and variance must still be float32 if the input is float16.

PiperOrigin-RevId: 170239448
parent 20370104
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