View source on GitHub |
Returns the device name for a core in a replicated TPU computation.
tf.compat.v1.tpu.core(
num
)
num
: the virtual core number within each replica to which operators should
be assigned.A device name, suitable for passing to tf.device()
.