MultiDeviceIteratorGetNextFromShard

public final class MultiDeviceIteratorGetNextFromShard

Gets next element for the provided shard number.

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

List<Output<?>>
components()
Result of the get_next on the dataset.
static MultiDeviceIteratorGetNextFromShard
create(Scope scope, Operand<?> multiDeviceIterator, Operand<TInt32> shardNum, Operand<TInt64> incarnationId, List<Class<? extends TType>> outputTypes, List<Shape> outputShapes)
Factory method to create a class wrapping a new MultiDeviceIteratorGetNextFromShard operation.
Iterator<Operand<TType>>

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "MultiDeviceIteratorGetNextFromShard"

Public Methods

public List<Output<?>> components ()

Result of the get_next on the dataset.

public static MultiDeviceIteratorGetNextFromShard create (Scope scope, Operand<?> multiDeviceIterator, Operand<TInt32> shardNum, Operand<TInt64> incarnationId, List<Class<? extends TType>> outputTypes, List<Shape> outputShapes)

Factory method to create a class wrapping a new MultiDeviceIteratorGetNextFromShard operation.

Parameters
scope current scope
multiDeviceIterator A MultiDeviceIterator resource.
shardNum Integer representing which shard to fetch data for.
incarnationId Which incarnation of the MultiDeviceIterator is running.
outputTypes The type list for the return values.
outputShapes The list of shapes being produced.
Returns
  • a new instance of MultiDeviceIteratorGetNextFromShard

public Iterator<Operand<TType>> iterator ()