提交 2f9cf7cc 编写于 作者: K Kexin Zhao

address comments

上级 a6718797
......@@ -116,15 +116,15 @@ void prune_impl(const proto::ProgramDesc& input, proto::ProgramDesc* output,
for (const auto& op : *op_field) {
// add VarDescs of all input arguments for each OpDesc
auto& input_field = op.inputs();
for (auto& input : input_field) {
for (auto& arg : input.arguments()) {
for (auto& input_var : input_field) {
for (auto& arg : input_var.arguments()) {
*var_field->Add() = var_map[arg];
}
}
// add VarDescs of all output arguments for each OpDesc
auto& output_field = op.outputs();
for (auto& output : output_field) {
for (auto& arg : output.arguments()) {
for (auto& output_var : output_field) {
for (auto& arg : output_var.arguments()) {
*var_field->Add() = var_map[arg];
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册