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

"split to generic add PR"

上级 0ac79a38
......@@ -71,6 +71,7 @@ static void DeDuplicate(NetOp* net, std::unordered_se)
auto* net = new NetOp();
if (forwardOp.IsNetOp()) {
//! TODO(dzh)
std::unordered_map<std::string, int> dup_output;
std::unordered_map<std::string std::vector<int>> dup_output_ops;
const unsigned uniq_id_local = uniq_id;
......@@ -98,12 +99,12 @@ static void DeDuplicate(NetOp* net, std::unordered_se)
if (op_ptr->inputs_[i] == dup.first) {
// unique the duplicate name
op_ptr->inputs_[i] += std::to_string(uniq_id++);
// TODO(dzh): need a generic add op here
}
}
}
}
//! TODO(dzh)
} else {
//! TODO(fjy)
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.
先完成此消息的编辑!
想要评论请 注册