diff --git a/paddle/framework/block_desc.cc b/paddle/framework/block_desc.cc index 2dbf54301a54d12b9d9597f4c696ee8745b3a5cd..e71eb3cb51037e678b5dbd42f73c3524514181d8 100644 --- a/paddle/framework/block_desc.cc +++ b/paddle/framework/block_desc.cc @@ -27,7 +27,7 @@ VarDescBind *BlockDescBind::Var(const std::string &name) { return var; } -VarDescBind *BlockDescBind::Var(const std::string &name) const { +VarDescBind *BlockDescBind::FindVar(const std::string &name) const { auto it = vars_.find(name); PADDLE_ENFORCE(it != vars_.end(), "Can not find variable %s in current block.", name); diff --git a/paddle/framework/block_desc.h b/paddle/framework/block_desc.h index 18af157b668bd56d20dc1c026e2d8e177059fc5e..1dd18a902e97dd4146c4f1fce568635f420a36b2 100644 --- a/paddle/framework/block_desc.h +++ b/paddle/framework/block_desc.h @@ -51,7 +51,7 @@ class BlockDescBind { VarDescBind *Var(const std::string &name_bytes); - VarDescBind *Var(const std::string &name_bytes) const; + VarDescBind *FindVar(const std::string &name_bytes) const; bool HasVar(const std::string &var_name) const; diff --git a/paddle/framework/operator.h b/paddle/framework/operator.h index 97a142d5f1661704fede858b28ff0d5487c66fab..cf15f9933ab3bc881add3d45b7ca17194a70e0f1 100644 --- a/paddle/framework/operator.h +++ b/paddle/framework/operator.h @@ -403,11 +403,11 @@ class CompileTimeInferShapeContext : public InferShapeContext { private: DDim GetDim(const std::string& name) const override { - return framework::make_ddim(block_.Var(name)->Shape()); + return framework::make_ddim(block_.FindVar(name)->Shape()); } void SetDim(const std::string& name, const DDim& dim) override { - block_.Var(name)->SetShape(framework::vectorize(dim)); + block_.FindVar(name)->SetShape(framework::vectorize(dim)); } const OpDescBind& op_;