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
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment