Added pybindings for dataformat and get_ranked_nodes

python_binding/data/pybind_Data.cpp

@@ -53,6 +53,30 @@ void init_Data(py::module& m){
     e_dtype.def("__str__", [enum_names](const DataType& dtype) {
         return enum_names[static_cast<int>(dtype)];
     }, py::prepend());;
+    // TODO : extend with more values !
+    py::enum_<DataType>(m, "dtype")
+    .value("float64", DataType::Float64)
+    .value("float32", DataType::Float32)
+    .value("float16", DataType::Float16)
+    .value("int8", DataType::Int8)
+    .value("int16", DataType::Int16)
+    .value("int32", DataType::Int32)
+    .value("int64", DataType::Int64)
+    .value("uint8", DataType::UInt8)
+    .value("uint16", DataType::UInt16)
+    .value("uint32", DataType::UInt32)
+    .value("uint64", DataType::UInt64)
+    ;
+
+    py::enum_<DataFormat>(m, "dformat")
+    .value("Default", DataFormat::Default)
+    .value("NCHW", DataFormat::NCHW) // default
+    .value("NHWC", DataFormat::NHWC) 
+    .value("CHWN", DataFormat::CHWN)
+    .value("NCDHW", DataFormat::NCDHW) 
+    .value("NDHWC", DataFormat::NDHWC)
+    .value("CDHWN", DataFormat::CDHWN)
+    ;
 
     py::class_<Data, std::shared_ptr<Data>>(m,"Data");
 

python_binding/data/pybind_Tensor.cpp

@@ -322,6 +322,7 @@ void init_Tensor(py::module& m){
     .def("grad", &Tensor::grad)
     .def("set_grad", &Tensor::setGrad)
     .def("dtype", &Tensor::dataType)
+    .def("dformat", &Tensor::dataFormat)
     .def("size", &Tensor::size)
     .def("capacity", &Tensor::capacity)
     .def("resize", (void (Tensor::*)(const std::vector<DimSize_t>&, std::vector<DimSize_t>)) &Tensor::resize, py::arg("dims"), py::arg("strides") = std::vector<DimSize_t>())

python_binding/graph/pybind_GraphView.cpp

@@ -142,6 +142,9 @@ void init_GraphView(py::module& m) {
           //                return py::none();
           //           }
           //      })
+          .def("get_ranked_nodes", &GraphView::getRankedNodes)
+          .def("set_dataformat", &GraphView::setDataFormat, py::arg("dataformat"))
+          
             ;
 
      m.def("get_connected_graph_view", &getConnectedGraphView);