Initializes the multi device iterator with the given dataset.
tf.raw_ops.MultiDeviceIteratorInit(
dataset, multi_device_iterator, max_buffer_size, name=None
)
Args | |
|---|---|
dataset
|
A Tensor of type variant. Dataset to be iterated upon.
|
multi_device_iterator
|
A Tensor of type resource.
A MultiDeviceIteratorResource.
|
max_buffer_size
|
A Tensor of type int64.
The maximum size of the host side per device buffer to keep.
|
name
|
A name for the operation (optional). |
Returns | |
|---|---|
A Tensor of type int64.
|