Searched refs:_create_chunk_sharded_tensor (Results 1 – 5 of 5) sorted by relevance
20 from torch.distributed.fsdp._shard_utils import _create_chunk_sharded_tensor157 inner_st = _create_chunk_sharded_tensor(185 inner_st = _create_chunk_sharded_tensor(211 return _create_chunk_sharded_tensor(
7 _create_chunk_sharded_tensor,33 sharded_tensor = _create_chunk_sharded_tensor(
11 _create_chunk_sharded_tensor,128 else _create_chunk_sharded_tensor
30 def _create_chunk_sharded_tensor( function
39 from torch.distributed.fsdp._shard_utils import _create_chunk_sharded_tensor303 state_dict[key] = _create_chunk_sharded_tensor(