Searched refs:_get_host_for_device (Results 1 – 1 of 1) sorted by relevance
243 host_device = _get_host_for_device(tpu_device)648 def _get_host_for_device(device): function