Commit 192b618b authored by Reed Wanderman-Milne's avatar Reed Wanderman-Milne Committed by TensorFlower Gardener
Browse files

Fix fp16 BatchNormalization when fused=False. 

Also fix BatchNormalization fp16 test. Before, the test was actually running in fp32, because the Keras would cast the input to the model to the input layer's dtype, which defaults to fp32.

PiperOrigin-RevId: 236133280
parent 68f2b667
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