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

fix review comments

parent 492d329a
...@@ -447,8 +447,7 @@ struct onnx_parser ...@@ -447,8 +447,7 @@ struct onnx_parser
s.visit([&](auto v) { copy(v, std::back_inserter(op.dims)); }); s.visit([&](auto v) { copy(v, std::back_inserter(op.dims)); });
} }
args[0] = make_contiguous(args[0]); return prog.add_instruction(op, make_contiguous(args[0]));
return prog.add_instruction(op, args[0]);
} }
instruction_ref instruction_ref
...@@ -496,11 +495,9 @@ struct onnx_parser ...@@ -496,11 +495,9 @@ struct onnx_parser
{ {
axis = parse_value(attributes.at("axis")).at<int>(); axis = parse_value(attributes.at("axis")).at<int>();
} }
args[0] = make_contiguous(args[0]);
args[1] = make_contiguous(args[1]);
op::gather op{axis}; 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 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