Searched defs:ProxyTorchDispatchMode (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | flex_attention.py | 292 proxy_mode: ProxyTorchDispatchMode,
360 mode: ProxyTorchDispatchMode,
832 proxy_mode: ProxyTorchDispatchMode,
916 mode: ProxyTorchDispatchMode,
|
| H A D | out_dtype.py | 141 mode: ProxyTorchDispatchMode,
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | proxy_tensor.py | 735 func: object, proxy_mode: ProxyTorchDispatchMode
761 proxy_mode: ProxyTorchDispatchMode,
1261 class ProxyTorchDispatchMode(TorchDispatchMode): class
1834 prev_proxy_mode: Union[nullcontext, ProxyTorchDispatchMode], argument
2163 proxy_mode: ProxyTorchDispatchMode,
|