Searched refs:has_kernel_for_dispatch_key (Results 1 – 4 of 4) sorted by relevance
93 def has_kernel_for_dispatch_key(self, k): member in OperatorBase179 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 …]
62 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)
65 if not any(torch_op.has_kernel_for_dispatch_key(k) for k in keys):
959 and op.has_kernel_for_dispatch_key(1248 return not op.has_kernel_for_dispatch_key(