Save the date! Google I/O returns May 18-20 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.