Searched refs:_dtensor_singleton (Results 1 – 1 of 1) sorted by relevance
37 _dtensor_singleton = None variable559 global _dtensor_singleton560 _dtensor_singleton = device565 if _dtensor_singleton is None:567 return _dtensor_singleton571 global _dtensor_singleton572 if _dtensor_singleton is not None:573 _dtensor_singleton.clear_tpu_core_ids()575 _dtensor_singleton = None