Searched refs:KernelRegistration (Results 1 – 1 of 1) sorted by relevance
1119 struct KernelRegistration { struct1120 KernelRegistration(const KernelDef& d, StringPiece c, in KernelRegistration() argument1134 std::unordered_multimap<string, KernelRegistration> registry1275 KernelRegistration(*kernel_def, kernel_class_name, std::move(factory))); in InitInternal()1309 StringPiece node_op, AttrSlice node_attrs, const KernelRegistration** reg, in FindKernelRegistration()1386 const KernelRegistration** reg, in FindKernelRegistration()1398 const KernelRegistration* reg = nullptr; in KernelDefAvailable()1412 const KernelRegistration* reg = nullptr; in FindKernelDef()1466 const KernelRegistration* reg = nullptr; in SupportedDeviceTypesForNode()1618 const KernelRegistration* registration = nullptr; in CreateOpKernel()