Commit 551d1f97 authored by Sergio Guadarrama's avatar Sergio Guadarrama Committed by TensorFlower Gardener
Browse files

Use nest.map_structure_up_to and nest.flatten_up_to directly. 

PiperOrigin-RevId: 234736064
parent 95147fbb
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