Searched refs:tensor_hooks (Results 1 – 1 of 1) sorted by relevance
86 tensor_hooks=None, argument93 self.tensor_hooks = tensor_hooks or {}105 and self.tensor_hooks == other.tensor_hooks125 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 …]