Home
last modified time | relevance | path

Searched defs:is_sharded (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/distributed/tensor/
D_dtensor_spec.py259 def is_sharded(self) -> bool: member in DTensorSpec
/external/pytorch/torch/distributed/_composable/fsdp/
D_fsdp_param_group.py471 def is_sharded(self) -> bool: member in FSDPParamGroup
/external/pytorch/torch/distributed/fsdp/
D_flat_param.py2419 def is_sharded(self, tensor: Tensor) -> bool: member in FlatParamHandle