Searched defs:func_overload (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/executorch/exir/ |
| H A D | delegate.py | 45 def trace_call_delegate(proxy_mode, func_overload, lowered_module, *args): argument
|
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| 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 | 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/pytorch/benchmarks/dynamo/microbenchmarks/ |
| H A D | operator_inp_utils.py | 172 def __torch_dispatch__(self, func_overload, types, args=(), kwargs=None): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_dynamic_shapes.py | 145 def __torch_dispatch__(cls, func_overload, types, args=(), kwargs=None): argument
|