"profiler/vscode:/vscode.git/clone" did not exist on "76d144fa7c396e52631719f79008e7099b6cd30d"
Commit b466ceb9 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 16b85d8b
...@@ -778,7 +778,8 @@ struct gather ...@@ -778,7 +778,8 @@ struct gather
{ {
argument result{output_shape}; argument result{output_shape};
// negative axis means counting dimensions from back // negative axis means counting dimensions from back
int axis_index = (axis < 0) ? static_cast<int>(args[0].get_shape().lens().size() + axis) : axis; int axis_index =
(axis < 0) ? static_cast<int>(args[0].get_shape().lens().size() + axis) : axis;
// max dimension in axis // max dimension in axis
visit_all(result, args[0])([&](auto output, auto data) { visit_all(result, args[0])([&](auto output, auto data) {
...@@ -789,13 +790,13 @@ struct gather ...@@ -789,13 +790,13 @@ struct gather
} }
else else
{ {
auto out_lens = data.get_shape().lens(); auto out_lens = data.get_shape().lens();
out_lens[axis_index] = indices.get_shape().elements(); out_lens[axis_index] = indices.get_shape().elements();
migraphx::shape out_comp_shape{data.get_shape().type(), out_lens}; migraphx::shape out_comp_shape{data.get_shape().type(), out_lens};
shape_for_each(out_comp_shape, [&](const auto& out_idx) { shape_for_each(out_comp_shape, [&](const auto& out_idx) {
auto data_idx = out_idx; auto data_idx = out_idx;
data_idx[axis_index] = indices[data_idx[axis_index]]; data_idx[axis_index] = indices[data_idx[axis_index]];
output[out_comp_shape.index(out_idx.begin(), out_idx.end())] = output[out_comp_shape.index(out_idx.begin(), out_idx.end())] =
data(data_idx.begin(), data_idx.end()); data(data_idx.begin(), data_idx.end());
}); });
} }
......
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