Gets next element for the provided shard number.
tf.raw_ops.MultiDeviceIteratorGetNextFromShard(
multi_device_iterator, shard_num, incarnation_id, output_types, output_shapes,
name=None
)
Args |
|---|
multi_device_iterator
|
A Tensor of type resource.
A MultiDeviceIterator resource.
|
shard_num
|
A Tensor of type int32.
Integer representing which shard to fetch data for.
|
incarnation_id
|
A Tensor of type int64.
Which incarnation of the MultiDeviceIterator is running.
|
output_types
|
A list of tf.DTypes that has length >= 1.
The type list for the return values.
|
output_shapes
|
A list of shapes (each a tf.TensorShape or list of ints) that has length >= 1.
The list of shapes being produced.
|
name
|
A name for the operation (optional).
|
Returns |
|---|
A list of Tensor objects of type output_types.
|
