Commit c04396e3 authored by Priya Gupta's avatar Priya Gupta Committed by TensorFlower Gardener
Browse files

Implement new API for TPUStrategy to run multiple steps, and move most of the... 

Implement new API for TPUStrategy to run multiple steps, and move most of the TPU specific logic into this method from `call_for_each_tower`.
Disable TPU tests temporarily, will enable again in subsequent code changes.

PiperOrigin-RevId: 201279470
parent 9751540a
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