Searched refs:_current_tpu_context (Results 1 – 1 of 1) sorted by relevance
43 _current_tpu_context = TpuContext() variable49 if _current_tpu_context.number_of_shards is not None:57 _current_tpu_context.set_number_of_shards(number_of_shards)60 _current_tpu_context.set_number_of_shards(None)64 return _current_tpu_context