Commit 2f77d96f authored by Justin Lebar's avatar Justin Lebar Committed by TensorFlower Gardener
Browse files

[XLA] Run ZeroSizedHloElimination after BatchNormExpander. 

BatchNormExpander can create zero-sized HLOs even if it's run after
ZeroSizedHloElimination.  For example, you might have a
BatchNormTraining op with shape (f32[0, 10], f32[10], f32[10]).  This
won't be eliminated by ZeroSizedHloElimination, but when it's
decomposed, it will result in many zero-sized HLOs.

PiperOrigin-RevId: 179885417
parent 6a9a9ed0
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