Tune in to the first Women in ML Symposium this Tuesday, October 19 at 9am PST 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.