Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A DOperatorEntry.cpp214 bool OperatorEntry::hasKernelForAnyDispatchKey(DispatchKeySet ks) const { in registerKernel() function in c10::impl::OperatorEntry
322 hasKernelForAnyDispatchKey(getBackendKeySetFromAutograd(dispatch_key)) || in registerKernel()
348 && hasKernelForAnyDispatchKey(c10::autogradother_backends)) { in registerKernel()
H A DDispatcher.h410 bool hasKernelForAnyDispatchKey(DispatchKeySet k) const { in hasKernelForAnyDispatchKey() function
411 return operatorDef_->op.hasKernelForAnyDispatchKey(k); in hasKernelForAnyDispatchKey()
H A DOperatorEntry.h203 bool hasKernelForAnyDispatchKey(DispatchKeySet ks) const;
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp599 return op->hasKernelForAnyDispatchKey(ks); in initDispatchBindings()