Searched refs:bind_tensors_to_proxies (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | compiled_autograd.py | 107 self.bind_tensors_to_proxies(inputs, args_proxy) 118 self.bind_tensors_to_proxies(sizes, sizes_proxy) 140 self.bind_tensors_to_proxies(scalars, scalars_proxy) 182 self.bind_tensors_to_proxies(grad_ins, proxies) 206 self.bind_tensors_to_proxies([inputs[i]], [proxy]) 219 self.bind_tensors_to_proxies(inputs, proxies) 233 self.bind_tensors_to_proxies(outputs, proxies) 247 self.bind_tensors_to_proxies(input, [proxy]) 450 def bind_tensors_to_proxies(self, tensors, proxies): member in AutogradCompilerInstance
|