Commit 9dc20c7c authored by Saurabh Saxena's avatar Saurabh Saxena Committed by TensorFlower Gardener
Browse files

Support taking gradients of de-serialized cond. 

Instead of relying on the _FuncGraphs attached to the op we instead reconstruct the _FuncGraph from the FunctionDef using function_def_to_graph.

PiperOrigin-RevId: 199525030
parent 8f2e5f0b
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