Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 5e378724 编写于 作者: F fengjiayi
浏览文件
操作

Refine code

上级 ab18947e
显示空白变更内容
内联 并排
Showing with 2 addition and 4 deletion
paddle/framework/grad_op_builder.cc
浏览文件 @ 5e378724
...@@ -45,12 +45,10 @@ static void TransOpArg(const OperatorBase* src_op, OperatorBase* dst_op, ...@@ -45,12 +45,10 @@ static void TransOpArg(const OperatorBase* src_op, OperatorBase* dst_op,
int& idx, bool is_grad) { int& idx, bool is_grad) {
const std::vector<std::string>& src_inout = const std::vector<std::string>& src_inout =
src_type == IN ? src_op->inputs_ : src_op->outputs_; src_type == IN ? src_op->inputs_ : src_op->outputs_;
const VarIndexMap& src_varmap = *src_op->in_out_idxs_;
const std::vector<int>* src_format = GetOpFormat(src_op, src_type); const std::vector<int>* src_format = GetOpFormat(src_op, src_type);
std::vector<std::string>& dst_inout = std::vector<std::string>& dst_inout =
dst_type == IN ? dst_op->inputs_ : dst_op->outputs_; dst_type == IN ? dst_op->inputs_ : dst_op->outputs_;
VarIndexMap& dst_varmap = *dst_op->in_out_idxs_;
std::vector<int>* dst_format = GetOpFormat(dst_op, dst_type); std::vector<int>* dst_format = GetOpFormat(dst_op, dst_type);
const OpProto& proto = OpRegistry::protos().at(src_op->type_); const OpProto& proto = OpRegistry::protos().at(src_op->type_);
const auto& src_arg_list = src_type == IN ? proto.inputs() : proto.outputs(); const auto& src_arg_list = src_type == IN ? proto.inputs() : proto.outputs();
...@@ -59,8 +57,8 @@ static void TransOpArg(const OperatorBase* src_op, OperatorBase* dst_op, ...@@ -59,8 +57,8 @@ static void TransOpArg(const OperatorBase* src_op, OperatorBase* dst_op,
std::string src_name = arg.name(); std::string src_name = arg.name();
std::string dst_name = std::string dst_name =
is_grad ? src_name + OperatorBase::GRAD_VAR_SUFFIX() : src_name; is_grad ? src_name + OperatorBase::GRAD_VAR_SUFFIX() : src_name;
dst_varmap[dst_name] = idx++; (*dst_op->in_out_idxs_)[dst_name] = idx++;
int src_arg_idx = src_varmap.at(src_name); int src_arg_idx = src_op->in_out_idxs_->at(src_name);
int src_begin = int src_begin =
src_format == nullptr ? src_arg_idx : src_format->at(src_arg_idx); src_format == nullptr ? src_arg_idx : src_format->at(src_arg_idx);
int src_end = src_format == nullptr ? src_arg_idx + 1 int src_end = src_format == nullptr ? src_arg_idx + 1
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或