From b8c5fdbf814cbc502b3271caac5b4248401bf28f Mon Sep 17 00:00:00 2001 From: Abhinav Arora Date: Wed, 14 Feb 2018 13:35:51 -0800 Subject: [PATCH] Fix protobuf.cc --- paddle/fluid/pybind/protobuf.cc | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/paddle/fluid/pybind/protobuf.cc b/paddle/fluid/pybind/protobuf.cc index 1b8fc1698b..4d3c1f280b 100644 --- a/paddle/fluid/pybind/protobuf.cc +++ b/paddle/fluid/pybind/protobuf.cc @@ -196,13 +196,13 @@ void BindBlockDesc(py::module &m) { void BindVarDsec(py::module &m) { py::enum_(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); + .value("BOOL", proto::VarType::BOOL) + .value("INT16", proto::VarType::INT16) + .value("INT32", proto::VarType::INT32) + .value("INT64", proto::VarType::INT64) + .value("FP16", proto::VarType::FP16) + .value("FP32", proto::VarType::FP32) + .value("FP64", proto::VarType::FP64); py::class_ var_desc(m, "VarDesc", ""); var_desc -- GitLab