tf.raw_ops.MultiDeviceIteratorGetNextFromShard

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.