Searched refs:kernel_creator (Results 1 – 3 of 3) sorted by relevance
40 …(const std::string &kernel_name, const KernelAttr &kernel_attr, CPUKernelCreator &&kernel_creator);57 …const std::string &kernel_name, const KernelAttr &kernel_attr, CPUKernelCreator &&kernel_creator) { in CPUKernelRegistrar() argument58 CPUKernelFactory::GetInstance().Register(kernel_name, kernel_attr, std::move(kernel_creator)); in CPUKernelRegistrar()
38 CPUKernelCreator &&kernel_creator) { in Register() argument39 (void)name_to_attr_creator_[kernel_name].emplace_back(kernel_attr, kernel_creator); in Register()
138 auto kernel_creator = GetCreator(key); in SupportKernel() local139 return kernel_creator != nullptr; in SupportKernel()