Commit 4e53b2c5 authored by Rohan Jain's avatar Rohan Jain Committed by TensorFlower Gardener
Browse files

Adding an _eager_reset() method to MultiDeviceIterator that re-initializes the... 

Adding an _eager_reset() method to MultiDeviceIterator that re-initializes the MultiDeviceIteratorResource and all underlying AnonymousIterator resources instead of re-creating them in eager mode as is the usual suggestion.

PiperOrigin-RevId: 234914867
parent 70bf6859
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