Searched refs:OpRegistration (Results 1 – 2 of 2) sorted by relevance
204 struct OpRegistration { struct243 static bool IsCompatible(const OpRegistration& x, const OpRegistration& y); argument253 std::unordered_map<string, std::vector<std::unique_ptr<OpRegistration>>> ops_311 std::unique_ptr<XlaOpRegistry::OpRegistration> Build(317 std::unique_ptr<XlaOpRegistry::OpRegistration> registration_;329 XlaOpRegistrar(std::unique_ptr<XlaOpRegistry::OpRegistration> registration);
61 /* static */ bool XlaOpRegistry::IsCompatible(const OpRegistration& x, in IsCompatible()62 const OpRegistration& y) { in IsCompatible()190 std::vector<std::unique_ptr<OpRegistration>>& op_registrations = ops.second; in RegisterCompilationKernels()194 [](const std::unique_ptr<OpRegistration>& op_reg) { in RegisterCompilationKernels()461 registration_.reset(new XlaOpRegistry::OpRegistration); in XlaOpRegistrationBuilder()531 std::unique_ptr<XlaOpRegistry::OpRegistration> XlaOpRegistrationBuilder::Build( in Build()538 std::unique_ptr<XlaOpRegistry::OpRegistration> registration) { in XlaOpRegistrar()