Searched defs:rank0_only (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/distributed/fsdp/ | ||
| D | api.py | 325 rank0_only: bool = False variable in FullStateDictConfig |
| /external/pytorch/test/distributed/tensor/parallel/ | ||
| D | test_parallelize_api.py | 41 rank0_only, argument |