Commit 1bf52f4c authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

A few micro optimizations to common sub-expression elimination in ArithmeticOptimizer: 

* Memoize node signatures.
* Don't discard candidate information after each outer iteration.
* Switch to using absl::flat_hash_map.
* Avoid copying regular inputs for non-commutative op types.

Add missing support for PrintV2.

PiperOrigin-RevId: 232030525
parent 4aae3817
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