Searched defs:op_overload (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | torchbind_impls.py | 127 def _register_py_impl_temporarily(op_overload, key, fn): argument
|
| /aosp_15_r20/external/pytorch/torch/_custom_op/ |
| H A D | autograd.py | 96 op_overload, argument
|
| /aosp_15_r20/external/pytorch/torch/_library/ |
| H A D | utils.py | 267 def handle_dispatch_mode(curr_mode, op_overload, *args, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_wrapper_cpu.py | 1989 self, op_overload, raw_args, output_args argument
2376 op_overload, argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | ir.py | 3955 op_overload: Optional[ variable in ExternKernel
3975 op_overload=None, argument
4565 op_overload=None, argument
4624 op_overload=None, argument
4827 def __init__(self, op_overload, x, *constant_args): argument
5003 op_overload, argument
5065 def __init__(self, op_overload, x, indices, values, accumulate): argument
|
| H A D | select_algorithm.py | 790 op_overload=None, argument
|
| H A D | lowering.py | 1807 def register_fallback(op_overload): argument
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | exported_program.py | 203 def assert_valid_to_preserve(op_overload): argument
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _ops.py | 632 def add_cached_op(op_overload): argument
|