Searched refs:KernelRegistration (Results 1 – 1 of 1) sorted by relevance
1106 struct KernelRegistration { struct1107 KernelRegistration(const KernelDef& d, StringPiece c, in KernelRegistration() argument1121 std::unordered_multimap<string, KernelRegistration> registry GUARDED_BY(mu);1262 KernelRegistration(*kernel_def, kernel_class_name, std::move(factory))); in InitInternal()1283 StringPiece node_op, AttrSlice node_attrs, const KernelRegistration** reg, in FindKernelRegistration()1360 const KernelRegistration** reg, in FindKernelRegistration()1372 const KernelRegistration* reg = nullptr; in KernelDefAvailable()1386 const KernelRegistration* reg = nullptr; in FindKernelDef()1433 const KernelRegistration* reg = nullptr; in SupportedDeviceTypesForNode()1566 const KernelRegistration* registration; in CreateOpKernel()