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

fixed protobuf.cc

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