Commit 1a4820e1 authored by Yanan Cao's avatar Yanan Cao Committed by TensorFlower Gardener
Browse files

TPUEstimator adds use_tpu key in params when calling model_fn. Model_fn can... 

TPUEstimator adds use_tpu key in params when calling model_fn. Model_fn can rely on this key to know whether TPU specific functionalities should be used.

PiperOrigin-RevId: 202542458
parent 15b258dc
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