fix a compile error

7 years ago · 0d03cab5e9
parent 7dabee2796
commit 0d03cab5e9
2 changed files with 2 additions and 2 deletions
--- a/paddle/framework/var_desc.cc
+++ b/paddle/framework/var_desc.cc
@ -56,7 +56,7 @@ size_t VarDesc::GetTensorDescNum() const {
 }

 void VarDesc::SetShapes(
-    const std::vector<const std::vector<int64_t>> &multiple_dims) {
+    const std::vector<std::vector<int64_t>> &multiple_dims) {
  PADDLE_ENFORCE_EQ(multiple_dims.size(), GetTensorDescNum(),
                    "The number of given shapes(%d) doesn't equal to the "
                    "number of sub tensor.",
--- a/paddle/framework/var_desc.h
+++ b/paddle/framework/var_desc.h
@ -74,7 +74,7 @@ class VarDesc {

  void SetShape(const std::vector<int64_t> &dims);

-  void SetShapes(const std::vector<const std::vector<int64_t>> &multiple_dims);
+  void SetShapes(const std::vector<std::vector<int64_t>> &multiple_dims);

  std::vector<int64_t> GetShape() const;