Searched refs:param_to_optim_hook_handle_map (Results 1 – 2 of 2) sorted by relevance
12 param_to_optim_hook_handle_map = torch.utils.weak.WeakTensorKeyDictionary() variable92 if param not in param_to_optim_hook_handle_map:93 param_to_optim_hook_handle_map[param] = []94 param_to_optim_hook_handle_map[param].append(handle)
1024 dist.optim.apply_optimizer_in_backward.param_to_optim_hook_handle_map