Searched refs:getKernelForDispatchKey (Results 1 – 2 of 2) sorted by relevance
245 const AnnotatedKernel* OperatorEntry::getKernelForDispatchKey(DispatchKey dispatch_key) const{ in registerKernel() function in c10::impl::OperatorEntry299 if (auto direct_registration = getKernelForDispatchKey(dispatch_key)) { in registerKernel()306 …if (auto default_backend_registration = getKernelForDispatchKey(DispatchKey::CompositeExplicitAuto… in registerKernel()314 …if (auto default_backend_registration = getKernelForDispatchKey(DispatchKey::CompositeExplicitAuto… in registerKernel()340 …if (auto nested_registration = getKernelForDispatchKey(DispatchKey::CompositeImplicitAutogradNeste… in registerKernel()346 if (auto math_registration = getKernelForDispatchKey(DispatchKey::CompositeImplicitAutograd)) { in registerKernel()358 if (auto autograd_registration = getKernelForDispatchKey(DispatchKey::Autograd)) { in registerKernel()366 …if (auto batched_registration = getKernelForDispatchKey(DispatchKey::FuncTorchBatchedDecomposition… in registerKernel()
309 const AnnotatedKernel* getKernelForDispatchKey(DispatchKey dispatch_key) const;