Searched refs:op_impl (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torch/_subclasses/ |
| D | fake_impls.py | 127 def impl_decorator(op_impl): argument
132 op_implementations_dict[run_impl_check] = op_impl
135 register_op_impl(op)(op_impl)
138 op_implementations_checks.append((run_impl_check, op_impl))
140 return op_impl
752 def impl_decorator(op_impl): argument
753 FAST_OP_IMPLEMENTATIONS[func] = op_impl
754 return op_impl
|
| D | fake_tensor.py | 1981 for run_impl_check, op_impl in op_implementations_checks:
1983 op_impl_out = op_impl(self, func, *args, **kwargs)
|
| /external/sdv/vsomeip/third_party/boost/mpl/doc/src/refmanual/ |
| D | NumericMetafunction.rst | 27 | op_impl< | | |
52 typedef op_impl<
|
| /external/pytorch/torch/ |
| D | __init__.py | 2509 def registerOp(cls, op_key, full_schema, op_impl, dispatch_key): argument
2512 cls.lib.impl("triton::" + op_key, op_impl, dispatch_key)
2513 cls.ops_table[(op_key, dispatch_key)] = op_impl
|
| /external/pytorch/test/ |
| D | test_ops.py | 2288 op_impl = getattr(import_module(f"torch.{module_path}"), op_name)
2292 op_impl,
2298 op_impl,
|
| /external/executorch/backends/qualcomm/utils/ |
| D | utils.py | 645 def op_impl(inputs: List[torch.Tensor]): function
|