View source on GitHub |
Returns the device name for a core in a replicated TPU computation.
tf.compat.v1.tpu.core(
num: int
) -> Text
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() .
|