提交 f78d7591 编写于 作者: F fengjiayi

Fix compile bug

上级 3003f53d
...@@ -21,7 +21,7 @@ void VarDescBind::SetShape(const std::vector<int64_t> &dims) { ...@@ -21,7 +21,7 @@ void VarDescBind::SetShape(const std::vector<int64_t> &dims) {
VectorToRepeated(dims, desc_.mutable_lod_tensor()->mutable_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); desc_.mutable_lod_tensor()->set_data_type(data_type);
} }
...@@ -29,7 +29,7 @@ std::vector<int64_t> VarDescBind::Shape() const { ...@@ -29,7 +29,7 @@ std::vector<int64_t> VarDescBind::Shape() const {
return RepeatedToVector(desc_.lod_tensor().dims()); return RepeatedToVector(desc_.lod_tensor().dims());
} }
DataType VarDescBind::DataType() const { DataType VarDescBind::GetDataType() const {
return desc_.lod_tensor().data_type(); return desc_.lod_tensor().data_type();
} }
} // namespace framework } // namespace framework
......
...@@ -64,7 +64,7 @@ class VarDescBind { ...@@ -64,7 +64,7 @@ class VarDescBind {
std::vector<int64_t> Shape() const; std::vector<int64_t> Shape() const;
DataType DataType() const; DataType GetDataType() const;
private: private:
VarDesc desc_; VarDesc desc_;
......
...@@ -167,7 +167,7 @@ void BindVarDsec(py::module &m) { ...@@ -167,7 +167,7 @@ void BindVarDsec(py::module &m) {
.def("set_shape", &VarDescBind::SetShape) .def("set_shape", &VarDescBind::SetShape)
.def("set_data_type", &VarDescBind::SetDataType) .def("set_data_type", &VarDescBind::SetDataType)
.def("shape", &VarDescBind::Shape, py::return_value_policy::reference) .def("shape", &VarDescBind::Shape, py::return_value_policy::reference)
.def("data_type", &VarDescBind::DataType); .def("data_type", &VarDescBind::GetDataType);
} }
void BindOpDesc(py::module &m) { void BindOpDesc(py::module &m) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册