| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | effects.py | 16 track_tensor_tree,
188 result = track_tensor_tree(out, out_proxy, constant=None, tracer=mode.tracer)
244 track_tensor_tree,
253 track_tensor_tree(token_tensor, token_proxy, constant=None, tracer=tracer)
|
| H A D | run_const_graph.py | 7 from torch.fx.experimental.proxy_tensor import ProxyTorchDispatchMode, track_tensor_tree
33 return track_tensor_tree(out, proxy, constant=None, tracer=mode.tracer)
|
| H A D | strict_mode.py | 14 track_tensor_tree,
77 return track_tensor_tree(out, out_proxy, constant=None, tracer=mode.tracer)
|
| H A D | torchbind.py | 12 from torch.fx.experimental.proxy_tensor import ProxyTorchDispatchMode, track_tensor_tree
106 ret = track_tensor_tree(out, out_proxy, constant=None, tracer=mode.tracer)
|
| H A D | hints_wrap.py | 15 from torch.fx.experimental.proxy_tensor import ProxyTorchDispatchMode, track_tensor_tree
141 return track_tensor_tree(out, out_proxy, constant=None, tracer=proxy_mode.tracer)
|
| H A D | out_dtype.py | 15 track_tensor_tree,
98 return track_tensor_tree(out, out_proxy, constant=None, tracer=proxy_mode.tracer)
|
| H A D | executorch_call_delegate.py | 23 track_tensor_tree,
70 return track_tensor_tree(out, out_proxy, constant=None, tracer=proxy_mode.tracer)
|
| H A D | while_loop.py | 18 from torch.fx.experimental.proxy_tensor import ProxyTorchDispatchMode, track_tensor_tree
219 return track_tensor_tree(
|
| H A D | map.py | 21 track_tensor_tree,
211 return track_tensor_tree(
|
| H A D | auto_functionalize.py | 17 track_tensor_tree,
609 result = track_tensor_tree(out, out_proxy, constant=None, tracer=mode.tracer)
704 result = track_tensor_tree(out, out_proxy, constant=None, tracer=mode.tracer)
|
| H A D | associative_scan.py | 24 track_tensor_tree,
333 return track_tensor_tree(out, out_proxy, constant=None, tracer=proxy_mode.tracer)
|
| H A D | cond.py | 34 track_tensor_tree,
340 return track_tensor_tree(out, out_proxy, constant=None, tracer=proxy_mode.tracer)
|
| H A D | flex_attention.py | 20 track_tensor_tree,
353 return track_tensor_tree(
909 return track_tensor_tree(
|
| H A D | triton_kernel_wrap.py | 22 track_tensor_tree,
605 ret = track_tensor_tree(out, out_proxy, constant=None, tracer=proxy_mode.tracer)
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | _trace_wrapped_higher_order_op.py | 8 from torch.fx.experimental.proxy_tensor import ProxyTorchDispatchMode, track_tensor_tree
102 track_tensor_tree(grad, out_proxy, constant=None, tracer=mode.tracer)
|
| H A D | compiled_autograd.py | 26 track_tensor_tree,
454 track_tensor_tree(tensors, proxies, constant=None, tracer=self.fx_tracer)
460 track_tensor_tree(bw_state, proxy, constant=None, tracer=self.fx_tracer)
|
| /aosp_15_r20/external/pytorch/torch/_prims/ |
| H A D | rng_prims.py | 15 track_tensor_tree,
213 return track_tensor_tree(out, out_proxy, constant=None, tracer=mode.tracer)
277 return track_tensor_tree(out, out_proxy, constant=None, tracer=mode.tracer)
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | delegate.py | 32 track_tensor_tree,
73 return track_tensor_tree(
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | wrappers.py | 11 from torch.fx.experimental.proxy_tensor import ProxyTorchDispatchMode, track_tensor_tree
32 return track_tensor_tree(args, proxy, constant=None, tracer=mode.tracer)
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | proxy_tensor.py | 592 def track_tensor_tree( function
969 track_tensor_tree(out, proxy_out, constant=constant, tracer=tracer)
1177 track_tensor_tree(flat_tensors, flat_proxies, constant=None, tracer=tracer)
1432 track_tensor_tree(out, proxy, constant=None, tracer=self.tracer)
1441 track_tensor_tree(out, proxy, constant=None, tracer=self.tracer)
|