[moco] Introduce GraphBuilderRegistry (#3226)

This will introduce GraphBuilderRegistry to manage GraphBuilders

Signed-off-by: SaeHie Park <saehie.park@samsung.com>

diff --git a/contrib/moco/lib/frontend/tf/src/Frontend.cpp b/contrib/moco/lib/frontend/tf/src/Frontend.cpp
index cd96cb2..0ba54ce 100644 (file)
--- a/contrib/moco/lib/frontend/tf/src/Frontend.cpp
+++ b/contrib/moco/lib/frontend/tf/src/Frontend.cpp
@@ -16,6 +16,9 @@
 
 #include <moco/tf/Frontend.h>
 
+#include "GraphBuilder.h"
+#include "GraphBuilderRegistry.h"
+
 #include <cwrap/Fildes.h>
 
 #include <tensorflow/core/framework/graph.pb.h>
@@ -89,6 +92,22 @@ void Frontend::load(const char *modelfile, FileType type) const
   load_tf(modelfile, type, tf_graph_def);
 
   // TODO convert tf_graph_def
+  for (const auto &n : tf_graph_def.node())
+  {
+    if (const auto *graph_builder = GraphBuilderRegistry::get().lookup(n.op()))
+    {
+      if (!graph_builder->validate(n))
+      {
+        throw std::runtime_error{"Invalid operator: " + n.op()};
+      }
+
+      // TODO call build
+    }
+    else
+    {
+      // TODO handle for not supported op
+    }
+  }
 }
 
 } // namespace tf

diff --git a/contrib/moco/lib/frontend/tf/src/GraphBuilderRegistry.h b/contrib/moco/lib/frontend/tf/src/GraphBuilderRegistry.h
new file mode 100644 (file)
index 0000000..48d2d1f
--- /dev/null
+++ b/contrib/moco/lib/frontend/tf/src/GraphBuilderRegistry.h
@@ -0,0 +1,68 @@
+/*
+ * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef __GRAPH_BUILDER_REGISTRY_H__
+#define __GRAPH_BUILDER_REGISTRY_H__
+
+#include "GraphBuilder.h"
+
+#include <map>
+#include <memory>
+#include <string>
+
+namespace moco
+{
+namespace tf
+{
+
+/**
+ * @brief Class to return graph builder for TF nodes
+ */
+class GraphBuilderRegistry
+{
+public:
+  /**
+   * @brief Returns registered GraphBuilder pointer for operator or
+   *        nullptr if not registered
+   */
+  const GraphBuilder *lookup(const std::string &op) const
+  {
+    if (_builder_map.find(op) == _builder_map.end())
+      return nullptr;
+
+    return _builder_map.at(op).get();
+  }
+
+  static GraphBuilderRegistry &get()
+  {
+    static GraphBuilderRegistry me;
+    return me;
+  }
+
+private:
+  GraphBuilderRegistry()
+  {
+    // TODO fill in GraphBuilders
+  }
+
+private:
+  std::map<std::string, std::unique_ptr<GraphBuilder>> _builder_map;
+};
+
+} // namespace tf
+} // namespace mono
+
+#endif // __GRAPH_BUILDER_REGISTRY_H__