Searched refs:_current_tpu_context (Results 1 – 1 of 1) sorted by relevance
42 _current_tpu_context = TpuContext() variable47 if _current_tpu_context.number_of_shards is not None:50 _current_tpu_context.set_number_of_shards(number_of_shards)53 _current_tpu_context.set_number_of_shards(None)57 return _current_tpu_context