提交 8a5ee462 编写于 作者: F fengjiayi

Fix some compile errors

上级 e192d0fd
...@@ -8,6 +8,8 @@ namespace framework { ...@@ -8,6 +8,8 @@ namespace framework {
class OpRegistry; class OpRegistry;
class GradOpCreator { class GradOpCreator {
using VarIndexMap = std::unordered_map<std::string, int>;
public: public:
GradOpCreator(const OperatorBase* op) : op_(op) {} GradOpCreator(const OperatorBase* op) : op_(op) {}
OperatorBase* Create(); OperatorBase* Create();
...@@ -32,15 +34,15 @@ class GradOpCreator { ...@@ -32,15 +34,15 @@ class GradOpCreator {
}; };
OpInOutArg* BuildArg(const VarProto& var, const VarIndexMap& var_map, OpInOutArg* BuildArg(const VarProto& var, const VarIndexMap& var_map,
const vector<int>& format, InOutType type); const std::vector<int>& format, InOutType type);
void BuildOpInOutArgList(); void BuildOpInOutArgList();
void PushArgIntoGradOp(const OpInOutArg* arg, vector<std::string>& in_out, void AddArgIntoGradOp(const OpInOutArg* arg, std::vector<std::string>& in_out,
vector<int>& format, VarIndexMap* varmap, int& idx, std::vector<int>& format, VarIndexMap* varmap, int& idx,
bool is_grad); bool is_grad);
void CompleteGradOp(OperatorBase* grad_op) const; void CompleteGradOp(OperatorBase* grad_op) const;
const OperatorBase* op_; const OperatorBase* op_;
std::vector<std::shared_ptr<OpInOutArg>> arg_list_; std::vector<std::shared_ptr<OpInOutArg>> arg_list_;
} };
} // namespace framework } // namespace framework
} // namespace paddle } // namespace paddle
...@@ -6,7 +6,7 @@ ...@@ -6,7 +6,7 @@
#include <unordered_map> #include <unordered_map>
#include <unordered_set> #include <unordered_set>
#include "paddle/framework/attr_checker.h" #include "paddle/framework/attr_checker.h"
#include "paddle/framework/grad_op_creater.h" #include "paddle/framework/grad_op_creator.h"
#include "paddle/framework/op_desc.pb.h" #include "paddle/framework/op_desc.pb.h"
#include "paddle/framework/scope.h" #include "paddle/framework/scope.h"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册