Commit 0c573ae9 authored by Tong Shen's avatar Tong Shen Committed by TensorFlower Gardener
Browse files

Skip SymbolicGradientOp when doing constant folding in control flow functionalization. 

If we want to evaluate SymbolicGradient op in constant folding, we need to construct Device object and attach it to FunctionLibraryRuntime. In graph rewriting pass, we do not have Device object created yet; it will only be created in XlaCompiler.

PiperOrigin-RevId: 214702943
parent 5d61748f
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