nn.Graph python

oneflow/api/python/framework/nn_graph.cpp

@@ -0,0 +1,30 @@
+/*
+Copyright 2020 The OneFlow Authors. 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.
+*/
+#include <pybind11/pybind11.h>
+#include <string>
+#include "oneflow/api/python/of_api_registry.h"
+#include "oneflow/core/framework/nn_graph_if.h"
+
+namespace py = pybind11;
+
+namespace oneflow {
+ONEFLOW_API_PYBIND11_MODULE("", m) {
+  using namespace oneflow;
+  py::class_<NNGraph, std::shared_ptr<NNGraph>>(m, "NNGraph")
+      .def(py::init<const std::string&>())
+      .def_property_readonly("name", &NNGraph::job_name);
+}
+}  // namespace oneflow

oneflow/core/framework/nn_graph.cpp

@@ -14,5 +14,11 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 #include "oneflow/core/framework/nn_graph_if.h"
+#include "oneflow/core/common/util.h"
+namespace oneflow {
 
-namespace oneflow {}
+const std::vector<std::string>& NNGraph::inputs_op_names() const { UNIMPLEMENTED(); }
+
+const std::vector<std::string>& NNGraph::outputs_op_names() const { UNIMPLEMENTED(); }
+
+}  // namespace oneflow

oneflow/core/framework/nn_graph_if.h

@@ -33,6 +33,19 @@ class NNGraphIf {
   NNGraphIf() = default;
 };
 
+class NNGraph final : public NNGraphIf {
+ public:
+  NNGraph() = delete;
+  explicit NNGraph(const std::string& name) : name_(name) {}
+
+  const std::string& job_name() const { return name_; }
+  const std::vector<std::string>& inputs_op_names() const;
+  const std::vector<std::string>& outputs_op_names() const;
+
+ private:
+  std::string name_;
+};
+
 }  // namespace oneflow
 
 #endif  // ONEFLOW_CORE_FRAMEWORK_NN_GRAPH_IF_H_