OrdinalSelector

public final class OrdinalSelector

A TPU core selector Op.

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.

Constants

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

Public Methods

Output<TInt32>
asOutput()
Returns the symbolic handle of the tensor.
static OrdinalSelector
create(Scope scope)
Factory method to create a class wrapping a new OrdinalSelector operation.
Output<TInt32>
deviceOrdinals()
A vector 1 or more TPU cores.

Inherited Methods

Constants

public static final String OP_NAME

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

Constant Value: "TPUOrdinalSelector"

Public Methods

public Output<TInt32> asOutput ()

Returns the symbolic handle of the tensor.

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.

public static OrdinalSelector create (Scope scope)

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

Parameters
scope current scope
Returns
  • a new instance of OrdinalSelector

public Output<TInt32> deviceOrdinals ()

A vector 1 or more TPU cores.