Searched refs:WeakTensorKeyDictionary (Results 1 – 7 of 7) sorted by relevance
12 param_to_optim_hook_handle_map = torch.utils.weak.WeakTensorKeyDictionary()13 param_to_acc_grad_map = torch.utils.weak.WeakTensorKeyDictionary()
10 from torch.utils.weak import WeakTensorKeyDictionary106 self.memo = WeakTensorKeyDictionary()
8 from torch.utils.weak import WeakTensorKeyDictionary12 _tensor_symint_registry = WeakTensorKeyDictionary()
32 from torch.utils.weak import WeakTensorKeyDictionary664 self.tensor_to_context = WeakTensorKeyDictionary()
303 WeakTensorKeyDictionary = WeakIdKeyDictionary variable
73 from torch.utils.weak import _WeakHashRef, WeakIdKeyDictionary, WeakTensorKeyDictionary1015 self.tensor_tracker = WeakTensorKeyDictionary()1398 self.tensor_tracker = WeakTensorKeyDictionary()
93 from torch.utils.weak import WeakTensorKeyDictionary10742 live_tensors = WeakTensorKeyDictionary()