Home
last modified time | relevance | path

Searched refs:has_kernel_for_dispatch_key (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/
H A D_ops.py93 def has_kernel_for_dispatch_key(self, k): member in OperatorBase
179 if op.has_kernel_for_dispatch_key(k):
185 ) and op.has_kernel_for_dispatch_key(cand):
191 ) and op.has_kernel_for_dispatch_key(cand):
195 ) or op.has_kernel_for_dispatch_key(DispatchKey.CompositeExplicitAutograd)
200 and op.has_kernel_for_dispatch_key(cand)
207 ) and op.has_kernel_for_dispatch_key(cand):
216 if is_included_in_alias(k, cand) and op.has_kernel_for_dispatch_key(cand):
220 if is_included_in_alias(k, cand) and op.has_kernel_for_dispatch_key(cand):
730 def has_kernel_for_dispatch_key(self, k): member in OpOverload
[all …]
/aosp_15_r20/external/executorch/exir/dialects/backend/
H A D_ops.py62 self._has_meta_kernel = self._op.has_kernel_for_dispatch_key(DispatchKey.Meta)
64 self._op.has_kernel_for_dispatch_key(DispatchKey.CompositeExplicitAutograd)
67 self._op.has_kernel_for_dispatch_key(DispatchKey.CompositeImplicitAutograd)
/aosp_15_r20/external/executorch/exir/dialects/
H A D_ops.py65 if not any(torch_op.has_kernel_for_dispatch_key(k) for k in keys):
/aosp_15_r20/external/pytorch/test/
H A Dtest_decomp.py959 and op.has_kernel_for_dispatch_key(
1248 return not op.has_kernel_for_dispatch_key(