提交 ec1d63af 编写于 作者: K Kavya Srinet

fixed protobuf.cc

上级 b14f84da
...@@ -195,14 +195,14 @@ void BindBlockDesc(py::module &m) { ...@@ -195,14 +195,14 @@ void BindBlockDesc(py::module &m) {
} }
void BindVarDsec(py::module &m) { void BindVarDsec(py::module &m) {
py::enum_<proto::DataType>(m, "DataType", "") py::enum_<proto::VarType::Type>(m, "DataType", "")
.value("BOOL", proto::DataType::BOOL) .value("BOOL", proto::VarType::Type::BOOL)
.value("INT16", proto::DataType::INT16) .value("INT16", proto::VarType::Type::INT16)
.value("INT32", proto::DataType::INT32) .value("INT32", proto::VarType::Type::INT32)
.value("INT64", proto::DataType::INT64) .value("INT64", proto::VarType::Type::INT64)
.value("FP16", proto::DataType::FP16) .value("FP16", proto::VarType::Type::FP16)
.value("FP32", proto::DataType::FP32) .value("FP32", proto::VarType::Type::FP32)
.value("FP64", proto::DataType::FP64); .value("FP64", proto::VarType::Type::FP64);
py::class_<VarDesc> var_desc(m, "VarDesc", ""); py::class_<VarDesc> var_desc(m, "VarDesc", "");
var_desc var_desc
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册