Commit ec55a682 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

fix review comments

parent 492d329a
......@@ -447,8 +447,7 @@ struct onnx_parser
s.visit([&](auto v) { copy(v, std::back_inserter(op.dims)); });
}
args[0] = make_contiguous(args[0]);
return prog.add_instruction(op, args[0]);
return prog.add_instruction(op, make_contiguous(args[0]));
}
instruction_ref
......@@ -496,11 +495,9 @@ struct onnx_parser
{
axis = parse_value(attributes.at("axis")).at<int>();
}
args[0] = make_contiguous(args[0]);
args[1] = make_contiguous(args[1]);
op::gather op{axis};
return prog.add_instruction(op, std::move(args));
return prog.add_instruction(op, make_contiguous(args[0]), make_contiguous(args[1]));
}
instruction_ref
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment