提交 f78d7591 编写于 作者: F fengjiayi

Fix compile bug

上级 3003f53d
......@@ -21,7 +21,7 @@ void VarDescBind::SetShape(const std::vector<int64_t> &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<int64_t> 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
......
......@@ -64,7 +64,7 @@ class VarDescBind {
std::vector<int64_t> Shape() const;
DataType DataType() const;
DataType GetDataType() const;
private:
VarDesc desc_;
......
......@@ -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) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册