提交 292f2ab1 编写于 作者: D dongzhihong

"split to generic add PR"

上级 0ac79a38
...@@ -71,6 +71,7 @@ static void DeDuplicate(NetOp* net, std::unordered_se) ...@@ -71,6 +71,7 @@ static void DeDuplicate(NetOp* net, std::unordered_se)
auto* net = new NetOp(); auto* net = new NetOp();
if (forwardOp.IsNetOp()) { if (forwardOp.IsNetOp()) {
//! TODO(dzh)
std::unordered_map<std::string, int> dup_output; std::unordered_map<std::string, int> dup_output;
std::unordered_map<std::string std::vector<int>> dup_output_ops; std::unordered_map<std::string std::vector<int>> dup_output_ops;
const unsigned uniq_id_local = uniq_id; const unsigned uniq_id_local = uniq_id;
...@@ -98,12 +99,12 @@ static void DeDuplicate(NetOp* net, std::unordered_se) ...@@ -98,12 +99,12 @@ static void DeDuplicate(NetOp* net, std::unordered_se)
if (op_ptr->inputs_[i] == dup.first) { if (op_ptr->inputs_[i] == dup.first) {
// unique the duplicate name // unique the duplicate name
op_ptr->inputs_[i] += std::to_string(uniq_id++); op_ptr->inputs_[i] += std::to_string(uniq_id++);
// TODO(dzh): need a generic add op here
} }
} }
} }
} }
//! TODO(dzh)
} else { } else {
//! TODO(fjy) //! TODO(fjy)
std::shared_ptr<OperatorBase> grad_op = OpRegistry::CreateGradOp(forwardOp); std::shared_ptr<OperatorBase> grad_op = OpRegistry::CreateGradOp(forwardOp);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册