Searched defs:wrap_fn (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/torch/_functorch/ |
| D | autograd_function.py | 113 def wrap_fn(output): function
168 outputs, unwrapped_inputs, orig_inputs, wrap_fn, out_dims=NO_OUT_DIMS argument
345 def wrap_fn(output, out_dim): function
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | multi_kernel.py | 304 def wrap_fn(kernel): function
|
| D | cpp.py | 3842 def wrap_fn(fn): function
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_device_type.py | 1757 def wrap_fn(self, *args, **kwargs): function
1792 def wrap_fn(self, *args, **kwargs): function
1810 def wrap_fn(self, *args, **kwargs): function
1828 def wrap_fn(self, *args, **kwargs): function
|
| D | common_utils.py | 1769 def wrap_fn(self, *args, **kwargs): function
1816 def wrap_fn(self, *args, **kwargs): function
|
| /external/libnl/python/netlink/ |
| D | core.py | 809 def wrap_fn(func): function
|
| /external/pytorch/test/inductor/ |
| D | test_cpu_repro.py | 1986 def wrap_fn(oh, ow): function
|