public final class TPUOrdinalSelector extends PrimitiveOp implements Operand<Integer>
This Op produces a set of TPU cores (for warm-up) or a single TPU core (for regular inference) to execute the TPU program on. The output is consumed by TPUPartitionedCall.
operation
Modifier and Type | Method and Description |
---|---|
Output<Integer> |
asOutput()
Returns the symbolic handle of a tensor.
|
static TPUOrdinalSelector |
create(Scope scope)
Factory method to create a class wrapping a new TPUOrdinalSelector operation.
|
Output<Integer> |
deviceOrdinals()
A vector 1 or more TPU cores.
|
equals, hashCode, op, toString
public static TPUOrdinalSelector create(Scope scope)
scope
- current scopepublic Output<Integer> asOutput()
Operand
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput
in interface Operand<Integer>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.