public class JointParallelDataSetIterator extends BaseParallelDataSetIterator
Modifier and Type | Class and Description |
---|---|
static class |
JointParallelDataSetIterator.Builder |
Modifier and Type | Field and Description |
---|---|
protected java.util.List<DataSetIterator> |
asyncIterators |
protected int |
bufferSizePerDevice |
protected boolean |
enforceSingleDevice |
allDepleted, counter, inequalityHandling, numProducers, producerAffinity, resetTracker, states
Constructor and Description |
---|
JointParallelDataSetIterator(java.util.List<DataSetIterator> iterators,
boolean singleDeviceMode,
int bufferSize,
InequalityHandling inequalityHandling) |
Modifier and Type | Method and Description |
---|---|
boolean |
hasNextFor(int consumer)
Returns true, if attached producer has something in queue, false otherwise
|
protected void |
initializeIterators(java.util.List<DataSetIterator> originals) |
DataSet |
nextFor(int consumer)
Returns next DataSet for given consumer
|
protected void |
reset(int consumer) |
asyncSupported, attachThread, batch, getCurrentProducerIndex, getLabels, getPreProcessor, hasNext, hasNextFor, inputColumns, next, next, nextFor, remove, reset, resetSupported, setPreProcessor, stepForward, totalOutcomes
protected java.util.List<DataSetIterator> asyncIterators
protected boolean enforceSingleDevice
protected int bufferSizePerDevice
public JointParallelDataSetIterator(@NonNull java.util.List<DataSetIterator> iterators, boolean singleDeviceMode, int bufferSize, @NonNull InequalityHandling inequalityHandling)
protected void initializeIterators(java.util.List<DataSetIterator> originals)
public boolean hasNextFor(int consumer)
ParallelDataSetIterator
hasNextFor
in interface ParallelDataSetIterator
hasNextFor
in class BaseParallelDataSetIterator
public DataSet nextFor(int consumer)
ParallelDataSetIterator
nextFor
in interface ParallelDataSetIterator
nextFor
in class BaseParallelDataSetIterator
protected void reset(int consumer)
reset
in class BaseParallelDataSetIterator