tf.raw_ops.TPUOrdinalSelector

A TPU core selector Op.

tf.raw_ops.TPUOrdinalSelector(
    name=None
)

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.

Args:

  • name: A name for the operation (optional).

Returns:

A Tensor of type int32.