org.tensorflow.op.core

Class MultiDeviceIteratorGetNextFromShard

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.MultiDeviceIteratorGetNextFromShard

All Implemented Interfaces:

Iterable<Operand<Object>>, Op

public final class MultiDeviceIteratorGetNextFromShard
extends PrimitiveOp
implements Iterable<Operand<Object>>

Gets next element for the provided shard number.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
List<Output<?>>	components() Result of the get_next on the dataset.
static MultiDeviceIteratorGetNextFromShard	create(Scope scope, Operand<?> multiDeviceIterator, Operand<Integer> shardNum, Operand<Long> incarnationId, List<Class<?>> outputTypes, List<Shape> outputShapes) Factory method to create a class to wrap a new MultiDeviceIteratorGetNextFromShard operation to the graph.
Iterator<Operand<Object>>	iterator()

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface java.lang.Iterable

forEach, spliterator

Method Detail

create

public static MultiDeviceIteratorGetNextFromShard create(Scope scope,
                                                         Operand<?> multiDeviceIterator,
                                                         Operand<Integer> shardNum,
                                                         Operand<Long> incarnationId,
                                                         List<Class<?>> outputTypes,
                                                         List<Shape> outputShapes)

Factory method to create a class to wrap a new MultiDeviceIteratorGetNextFromShard operation to the graph.

Parameters:

scope - current graph 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

components

public List<Output<?>> components()

Result of the get_next on the dataset.

iterator

public Iterator<Operand<Object>> iterator()

Specified by:

iterator in interface Iterable<Operand<Object>>