Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dside_effects.py86 tensor_hooks=None, argument
93 self.tensor_hooks = tensor_hooks or {}
105 and self.tensor_hooks == other.tensor_hooks
125 elif self.tensor_hooks != other.tensor_hooks:
139 tensor_hooks=self.tensor_hooks,
455 idx = len(self.tensor_hooks.keys())
457 while idx in self.tensor_hooks:
459 self.tensor_hooks[idx] = (tensor, hook, handle, name)
464 del self.tensor_hooks[idx]
472 ) in self.tensor_hooks.values():
[all …]