Commit 901a8ec9 authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

Support captures in tf.compat.v1.wrap_function.prune() 

Also sets the output structure instead of returning the flattened structure

PiperOrigin-RevId: 225911821
parent fc26aad8
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