Commit 834255a3 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 26ce057d
......@@ -784,8 +784,9 @@ struct gather
// max dimension in axis
visit_all(result, args[0])([&](auto output, auto data) {
std::vector<std::size_t> vec_indices;
args[1].visit([&](auto indices) { vec_indices.assign(indices.begin(), indices.end()); });
if (output_shape.scalar())
args[1].visit(
[&](auto indices) { vec_indices.assign(indices.begin(), indices.end()); });
if(output_shape.scalar())
{
output[0] = data[vec_indices.front()];
}
......@@ -797,7 +798,8 @@ struct gather
shape_for_each(out_comp_shape, [&](const auto& out_idx) {
auto data_idx = out_idx;
data_idx[axis_index] = vec_indices[data_idx[axis_index]];
//output(out_idx.begin(), out_idx.end()) = data(data_idx.begin(), data_idx.end());
// output(out_idx.begin(), out_idx.end()) = data(data_idx.begin(),
// data_idx.end());
output[out_comp_shape.index(out_idx)] = 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