Package | Description |
---|---|
org.nd4j.linalg.dataset | |
org.nd4j.linalg.dataset.callbacks |
Modifier and Type | Field and Description |
---|---|
protected DataSetCallback |
AsyncDataSetIterator.callback |
protected DataSetCallback |
AsyncMultiDataSetIterator.callback |
Constructor and Description |
---|
AsyncDataSetIterator(DataSetIterator iterator,
int queueSize,
BlockingQueue<DataSet> queue,
boolean useWorkspace,
DataSetCallback callback) |
AsyncDataSetIterator(DataSetIterator iterator,
int queueSize,
BlockingQueue<DataSet> queue,
boolean useWorkspace,
DataSetCallback callback,
Integer deviceId) |
AsyncDataSetIterator(DataSetIterator baseIterator,
int queueSize,
boolean useWorkspace,
DataSetCallback callback) |
AsyncMultiDataSetIterator(MultiDataSetIterator iterator,
int queueSize,
BlockingQueue<MultiDataSet> queue,
boolean useWorkspace,
DataSetCallback callback) |
AsyncMultiDataSetIterator(MultiDataSetIterator iterator,
int queueSize,
BlockingQueue<MultiDataSet> queue,
boolean useWorkspace,
DataSetCallback callback,
Integer deviceId) |
Modifier and Type | Class and Description |
---|---|
class |
DefaultCallback
This callback ensures that memory on device is up-to-date with host memory.
|
Copyright © 2019. All rights reserved.