Home
last modified time | relevance | path

Searched defs:_reset_flat_param_grad_info_if_needed (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/distributed/fsdp/
D_runtime_utils.py1550 def _reset_flat_param_grad_info_if_needed( function
D_flat_param.py2363 def _reset_flat_param_grad_info_if_needed(self): member in FlatParamHandle