o ?er@s>dZddlmZddlmZedgddedefdd Zd S) z&Helper functions for TPU device names.)Text) tf_exportztpu.core)v1numreturncCs d|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)rrd/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/tpu/tpu_name_util.pycores r N)__doc__typingrZ tensorflow.python.util.tf_exportrintr rrrr s