Searched refs:hasKernelForDispatchKey (Results 1 – 6 of 6) sorted by relevance
402 bool hasKernelForDispatchKey(DispatchKey k) const { in hasKernelForDispatchKey() function403 return operatorDef_->op.hasKernelForDispatchKey(k); in hasKernelForDispatchKey()758 if (op.hasKernelForDispatchKey(dk)) {
223 bool OperatorEntry::hasKernelForDispatchKey(DispatchKey k) const { in registerKernel() function in c10::impl::OperatorEntry324 hasKernelForDispatchKey(DispatchKey::CompositeExplicitAutograd); in registerKernel()
120 return op.hasKernelForDispatchKey(dk); in waitForImpl()500 if (!k || op.second.hasKernelForDispatchKey(*k)) { in getRegistrationsForDispatchKey()
205 bool hasKernelForDispatchKey(DispatchKey k) const;
115 if (triton_kernel.hasKernelForDispatchKey(c10::DispatchKey::SparseCsrCUDA)) { in _compressed_row_strided_mm_out()244 if (triton_kernel.hasKernelForDispatchKey(c10::DispatchKey::SparseCsrCUDA)) { in _compressed_row_strided_addmm_out()
580 return op->hasKernelForDispatchKey(dispatch); in initDispatchBindings()