TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.tpu.core

View source on GitHub

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

Aliases:

tf.tpu.core(num)

Args:

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

Returns:

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