Commit 8dacf726 authored by Russell Power's avatar Russell Power Committed by TensorFlower Gardener
Browse files

TPUEstimator: always force dequeue operations to execute. 

If a hostcall does not use the result of an outfeed_dequeue operation, the
resulting values remains on the outfeed.  This eventually stalls the training
process.  This CL executes all outfeed_dequeue operations explicitly when
running hostcalls to ensure the outfeed is drained.

PiperOrigin-RevId: 229812586
parent 793d6699
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