Google I/O returns May 18-20! Reserve space and build your schedule Register now


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.