ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


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.

name A name for the operation (optional).

A Tensor of type int32.