Searched refs:_TPU_DEVICE_REG (Results 1 – 1 of 1) sorted by relevance
37 _TPU_DEVICE_REG = re.compile(r'.*task:(\d+)/.*device:TPU:(\d+)$') variable104 match = _TPU_DEVICE_REG.match(device.name)