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

Add an API to distribution strategy that allows running N steps. Implement... 

Add an API to distribution strategy that allows running N steps. Implement this for MirroredStrategy and OneDeviceStrategy. Implemented in TPUStrategy earlier.

PiperOrigin-RevId: 207961939
parent 643d809b
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