Searched defs:reduce_dtype (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/distributed/_composable/fsdp/ | ||
| D | _fsdp_api.py | 45 reduce_dtype: Optional[torch.dtype] = None variable in MixedPrecisionPolicy |
| /external/pytorch/torch/distributed/fsdp/ | ||
| D | api.py | 221 reduce_dtype: Optional[torch.dtype] = None variable in MixedPrecision |
| /external/pytorch/torch/nn/parallel/ | ||
| D | distributed.py | 90 reduce_dtype: Optional[torch.dtype] = None variable in _MixedPrecision |