Home
last modified time | relevance | path

Searched refs:_ext_chunk_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D_fsdp_extensions.py117 def _ext_chunk_tensor( function
H A D_state_dict_utils.py57 _ext_chunk_tensor,
550 sharded_tensor = _ext_chunk_tensor(
H A D_optim_utils.py44 _ext_chunk_tensor,
319 optim_state = _ext_chunk_tensor(
1484 value = _ext_chunk_tensor(