diff --git a/paddle/framework/var_desc.cc b/paddle/framework/var_desc.cc index 1ccb81879ad293ad90cf64169952aa7f0f3df89a..13b9c5f3cdf98e6d22f4217fa1cf9a48910a78d8 100644 --- a/paddle/framework/var_desc.cc +++ b/paddle/framework/var_desc.cc @@ -21,7 +21,7 @@ void VarDescBind::SetShape(const std::vector &dims) { VectorToRepeated(dims, desc_.mutable_lod_tensor()->mutable_dims()); } -void VarDescBind::SetDataType(enum DataType data_type) { +void VarDescBind::SetDataType(DataType data_type) { desc_.mutable_lod_tensor()->set_data_type(data_type); } @@ -29,7 +29,7 @@ std::vector VarDescBind::Shape() const { return RepeatedToVector(desc_.lod_tensor().dims()); } -DataType VarDescBind::DataType() const { +DataType VarDescBind::GetDataType() const { return desc_.lod_tensor().data_type(); } } // namespace framework diff --git a/paddle/framework/var_desc.h b/paddle/framework/var_desc.h index 6384da90967503a7b4b4e7fe3f9f9dcb06301991..4763bf09d004539ab24e4aad3bf429667f1fcc73 100644 --- a/paddle/framework/var_desc.h +++ b/paddle/framework/var_desc.h @@ -64,7 +64,7 @@ class VarDescBind { std::vector Shape() const; - DataType DataType() const; + DataType GetDataType() const; private: VarDesc desc_; diff --git a/paddle/pybind/protobuf.cc b/paddle/pybind/protobuf.cc index 19ea26897f538ee3be611bb7629fc59b81c9c67a..218821b35bb6947181fedc56e002ad0285f6307d 100644 --- a/paddle/pybind/protobuf.cc +++ b/paddle/pybind/protobuf.cc @@ -167,7 +167,7 @@ void BindVarDsec(py::module &m) { .def("set_shape", &VarDescBind::SetShape) .def("set_data_type", &VarDescBind::SetDataType) .def("shape", &VarDescBind::Shape, py::return_value_policy::reference) - .def("data_type", &VarDescBind::DataType); + .def("data_type", &VarDescBind::GetDataType); } void BindOpDesc(py::module &m) {