Searched refs:KernelRegistration (Results 1 – 1 of 1) sorted by relevance
941 struct KernelRegistration { struct942 KernelRegistration(const KernelDef& d, StringPiece c, in KernelRegistration() argument954 typedef std::unordered_multimap<string, KernelRegistration> KernelRegistry;1085 ->emplace(key, KernelRegistration(*kernel_def, kernel_class_name, in InitInternal()1105 const KernelRegistration** reg, in FindKernelRegistration()1139 const KernelRegistration* reg = nullptr; in KernelDefAvailable()1149 const KernelRegistration* reg = nullptr; in FindKernelDef()1183 const KernelRegistration* reg = nullptr; in SupportedDeviceTypesForNode()1281 const KernelRegistration* registration; in CreateOpKernel()