BVhrJdZddlmZddlmZedgdedefdZy ) z&Helper functions for TPU device names.)Text) tf_exportztpu.core)v1numreturnc$dj|S)zReturns the device name for a core in a replicated TPU computation. 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()`. zdevice:TPU_REPLICATED_CORE:{})format)rs S/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/tpu/tpu_name_util.pycorer s ) / / 44N)__doc__typingr tensorflow.python.util.tf_exportrintr r r rs9-6 zl 5c 5d 5 5r