Searched defs:proxy_mode (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | hints_wrap.py | 121 def trace_hints_wrapper(proxy_mode, hints_wrapper, body_fn, args, kwargs, hints): argument
145 def inner(proxy_mode, body_fn, args, kwargs, hints): argument
|
| H A D | executorch_call_delegate.py | 46 def trace_call_delegate(proxy_mode, func_overload, lowered_module, *args): argument
|
| H A D | out_dtype.py | 80 def trace_out_dtype(proxy_mode, func_overload, op, output_dtype, *args): argument
|
| H A D | while_loop.py | 188 proxy_mode, while_loop_op, cond_fn, body_fn, carried_inputs, additional_inputs argument
|
| H A D | utils.py | 200 def unique_graph_id(proxy_mode, prefix): argument
|
| H A D | associative_scan.py | 282 proxy_mode, func_overload, combine_fn: Callable, input: List[torch.Tensor], dim: int argument
|
| H A D | map.py | 184 def trace_map(proxy_mode, func_overload, f, xs, pos_args): argument
|
| H A D | cond.py | 221 def trace_cond(proxy_mode, func_overload, pred, true_fn, false_fn, operands): argument
|
| H A D | triton_kernel_wrap.py | 593 def trace_triton_kernel_wrapper(proxy_mode, func_overload, node_args): argument
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | delegate.py | 45 def trace_call_delegate(proxy_mode, func_overload, lowered_module, *args): argument
|