Searched refs:OpRegistration (Results 1 – 2 of 2) sorted by relevance
248 struct OpRegistration { struct293 static bool IsCompatible(const OpRegistration& x, const OpRegistration& y); argument303 std::unordered_map<string, std::vector<std::unique_ptr<OpRegistration>>> ops_367 std::unique_ptr<XlaOpRegistry::OpRegistration> Build(373 std::unique_ptr<XlaOpRegistry::OpRegistration> registration_;385 XlaOpRegistrar(std::unique_ptr<XlaOpRegistry::OpRegistration> registration);
61 /* static */ bool XlaOpRegistry::IsCompatible(const OpRegistration& x, in IsCompatible()62 const OpRegistration& y) { in IsCompatible()201 std::vector<std::unique_ptr<OpRegistration>>& op_registrations = ops.second; in RegisterCompilationKernels()205 [](const std::unique_ptr<OpRegistration>& op_reg) { in RegisterCompilationKernels()482 registration_.reset(new XlaOpRegistry::OpRegistration); in XlaOpRegistrationBuilder()562 std::unique_ptr<XlaOpRegistry::OpRegistration> XlaOpRegistrationBuilder::Build( in Build()569 std::unique_ptr<XlaOpRegistry::OpRegistration> registration) { in XlaOpRegistrar()