Commit bee95569 authored by Eugene Brevdo's avatar Eugene Brevdo Committed by TensorFlower Gardener
Browse files

[TF Grappler] MetaOptimizer now takes a ConfigProto.

This will allow us to start reasoning about more global config options in
Grappler.  For example, it will allow us to move XLA functionalization rewrites
into Grappler, because we'll know when JIT optimization has been turned on.

We'll also be able to selectively disable particular grappler rewrites when
JIT optimization is on.

Additional changes:

* PartitionedCall now accepts a ConfigProto instead of a RewriterConfig
* Eager context now passes the entire ConfigProto through.

PiperOrigin-RevId: 221460335
parent 2a555cd9
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment