Home
last modified time | relevance | path

Searched refs:_tpu_system_device_name (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtpu_strategy_util.py99 with ops.device(tpu._tpu_system_device_name(job)): # pylint: disable=protected-access
127 with ops.device(tpu._tpu_system_device_name(job)): # pylint: disable=protected-access
195 with ops.device(tpu._tpu_system_device_name(job)): # pylint: disable=protected-access
Dtpu.py104 def _tpu_system_device_name(job: Optional[Text]) -> Text: function
136 with ops.device(_tpu_system_device_name(job)):
178 with ops.device(_tpu_system_device_name(job)):
192 with ops.device(_tpu_system_device_name(job)):