Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dcompiled_autograd.py22 disable_proxy_modes_tracing,
170 with disable_proxy_modes_tracing():
204 with disable_proxy_modes_tracing():
217 with disable_proxy_modes_tracing():
231 with disable_proxy_modes_tracing():
245 with disable_proxy_modes_tracing():
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dmap.py18 disable_proxy_modes_tracing,
73 with disable_proxy_modes_tracing():
196 with disable_proxy_modes_tracing():
250 with disable_proxy_modes_tracing():
H A Deffects.py14 disable_proxy_modes_tracing,
169 with disable_proxy_modes_tracing():
243 disable_proxy_modes_tracing,
247 with disable_proxy_modes_tracing():
H A Dassociative_scan.py22 disable_proxy_modes_tracing,
284 with disable_proxy_modes_tracing():
330 with disable_proxy_modes_tracing():
H A Dstrict_mode.py11 disable_proxy_modes_tracing,
62 with disable_proxy_modes_tracing():
H A Dout_dtype.py12 disable_proxy_modes_tracing,
88 with disable_proxy_modes_tracing():
H A Dexecutorch_call_delegate.py20 disable_proxy_modes_tracing,
60 with disable_proxy_modes_tracing():
H A Dauto_functionalize.py15 disable_proxy_modes_tracing,
599 with disable_proxy_modes_tracing():
694 with disable_proxy_modes_tracing():
H A Dcond.py32 disable_proxy_modes_tracing,
184 with disable_proxy_modes_tracing():
H A Dtriton_kernel_wrap.py20 disable_proxy_modes_tracing,
594 with disable_proxy_modes_tracing():
H A Dflex_attention.py489 from torch.fx.experimental.proxy_tensor import disable_proxy_modes_tracing
502 with disable_proxy_modes_tracing():
/aosp_15_r20/external/executorch/exir/
H A Ddelegate.py29 disable_proxy_modes_tracing,
59 with disable_proxy_modes_tracing():
/aosp_15_r20/external/pytorch/torch/_prims/
H A Drng_prims.py13 disable_proxy_modes_tracing,
270 with disable_proxy_modes_tracing():
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dproxy_tensor.py1175 with disable_proxy_modes_tracing() as m:
2151 with disable_proxy_modes_tracing():
2158 def disable_proxy_modes_tracing() -> Generator[ProxyTorchDispatchMode, None, None]: function
2193 with disable_proxy_modes_tracing():