From d8e391dd485f37d9269acf57cac88cf8fe74279f Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Tue, 9 Jul 2024 09:24:43 +0000
Subject: [PATCH] Add barious methods to python binding.
---
python_binding/data/pybind_Data.cpp | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/python_binding/data/pybind_Data.cpp b/python_binding/data/pybind_Data.cpp
index b1e879d83..13daf2198 100644
--- a/python_binding/data/pybind_Data.cpp
+++ b/python_binding/data/pybind_Data.cpp
@@ -60,6 +60,16 @@ void init_Data(py::module& m){
bindEnum<DataType>(m, "dtype");
bindEnum<DataFormat>(m, "dformat");
+ py::enum_<DataFormat>(m, "dformat")
+ .value("Default", DataFormat::Default)
+ .value("NCHW", DataFormat::NCHW)
+ .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");
--
GitLab