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

The GeneratorDataset init function was being run during Initialization which... 

The GeneratorDataset init function was being run during Initialization which is a blocking Op. Moving it to the GetNext call which is a non blocking async op.

This means we can now revert back to the default of 2 threads per GPU.

PiperOrigin-RevId: 214614839
parent 01512356
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