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

tf.compat.v1.tpu.core

View source on GitHub

Returns the device name for a core in a replicated TPU computation.

num the virtual core number within each replica to which operators should be assigned.

A device name, suitable for passing to tf.device().