Searched defs:_use_dtensor (Results 1 – 1 of 1) sorted by relevance
349 _use_dtensor: bool = False variable in ShardedStateDictConfig403 _use_dtensor: bool = False variable in ShardedOptimStateDictConfig