Searched refs:_tpu_init_fn (Results 1 – 2 of 2) sorted by relevance
97 def _tpu_init_fn(): function112 output = _tpu_init_fn()
190 def _tpu_init_fn(): function199 my_core_ids = _tpu_init_fn()