Searched refs:get_first_tpu_host_device (Results 1 – 2 of 2) sorted by relevance
37 def get_first_tpu_host_device(cluster_resolver): function91 with ops.device(get_first_tpu_host_device(cluster_resolver)):
233 self._host_device = tpu_strategy_util.get_first_tpu_host_device(