Home
last modified time | relevance | path

Searched refs:_get_param_to_param_key (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A Dfully_sharded_data_parallel.py95 _get_param_to_param_key,
1787 else _get_param_to_param_key(optim)
H A D_optim_utils.py903 else _get_param_to_param_key(
1120 def _get_param_to_param_key( function