Commit 428dbbfd authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by Benoit Steiner
Browse files

[BatchNormRewriter] Add option to use multi-output fusion. 

Having an option to opt-out multi-output fusion has two benefits:

- This enables CPU backend, which doesn't have fusion support, to handle the rewritten nodes.

- This helps our benchmarks to turn off multi-output fusion.

RELNOTES: n/a
PiperOrigin-RevId: 163990633
parent f3ecb017
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