Fix compile bug

update-doc-pybind
fengjiayi 8 years ago
parent 3003f53d52
commit f78d7591d2

@ -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) {

Loading…
Cancel
Save