Searched refs:call_torchbind (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | torchbind.py | 33 call_torchbind = CallTorchBind() variable
40 has_side_effect(call_torchbind)
47 return call_torchbind(self.raw_owner, self.name, *args, **kwargs)
68 @call_torchbind.py_impl(DispatchKey.CompositeExplicitAutograd)
78 @call_torchbind.py_impl(ProxyTorchDispatchMode)
85 call_torchbind,
89 out = call_torchbind(*args, **kwargs)
118 @call_torchbind.py_impl(FakeTensorMode)
131 call_torchbind.py_impl(DispatchKey.Autograd)(
132 autograd_not_implemented(call_torchbind, deferred_error=True)
[all …]
|
| H A D | effects.py | 10 from torch._higher_order_ops.torchbind import call_torchbind
30 call_torchbind: _EffectType.ORDERED,
199 elif op == call_torchbind:
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | script_object.py | 51 from torch._higher_order_ops.torchbind import call_torchbind
69 call_torchbind,
|
| H A D | builder.py | 37 from torch._higher_order_ops.torchbind import call_torchbind
2328 and proxy.node.target is call_torchbind
|
| /aosp_15_r20/external/pytorch/torch/_library/ |
| H A D | fake_class_registry.py | 33 from torch._higher_order_ops.torchbind import call_torchbind
35 return call_torchbind(self.self_fake_obj, self.method_name, *args, **kwargs)
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | _remove_effect_tokens_pass.py | 56 if func == torch.ops.higher_order.call_torchbind:
|