Commit 6c7e526e authored by Justin Lebar's avatar Justin Lebar Committed by TensorFlower Gardener
Browse files

[XLA] Add flag to BatchNormExpander pass that lets it use explicit broadcasts... 

[XLA] Add flag to BatchNormExpander pass that lets it use explicit broadcasts instead of kMap instructions.

PiperOrigin-RevId: 199867000
parent 426ea5b2
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