Project import generated by Copybara.

GitOrigin-RevId: 53a42bf7ad836321123cb7b6c80b0f2e13fbf83e
2020-04-06 19:14:13 -07:00
parent 1722d4b8a2
commit a3d36eee32
127 changed files with 3910 additions and 4783 deletions
@@ -94,6 +94,10 @@ class GraphRegistry {
  // Ownership of the specified FunctionRegistry is not transferred.
  GraphRegistry(FunctionRegistry<std::unique_ptr<Subgraph>>* factories);

+  // Registers a graph config builder type, using a factory function.
+  void Register(const std::string& type_name,
+                std::function<std::unique_ptr<Subgraph>()> factory);
+
  // Registers a graph config by name.
  void Register(const std::string& type_name,
                const CalculatorGraphConfig& config);