operation
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 wrapping a new MultiDeviceIteratorGetNextFromShard operation.
|
Iterator<Operand<Object>> |
iterator() |
equals, hashCode, op, toString
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
forEach, spliterator
public static MultiDeviceIteratorGetNextFromShard create(Scope scope, Operand<?> multiDeviceIterator, Operand<Integer> shardNum, Operand<Long> incarnationId, List<Class<?>> outputTypes, List<Shape> outputShapes)
scope
- current scopemultiDeviceIterator
- 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.Copyright © 2022. All rights reserved.