Commit 95312044 authored by Olivia Nordquist's avatar Olivia Nordquist Committed by TensorFlower Gardener
Browse files

Allows feature inputs to be arbitrary nested tuples in TPUEstimator which is... 

Allows feature inputs to be arbitrary nested tuples in TPUEstimator which is allowed behavior in  tf.Estimator

NB: Calls data/util/nest.flatten() which is slightly different than core/util/nest (already a dependency in tpu_estimator.py) as it maintains structure by treating tuples and lists differently
PiperOrigin-RevId: 208848267
parent df0c18e9
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