Skip to content

Commit

Permalink
Fix format
Browse files Browse the repository at this point in the history
  • Loading branch information
notoraptor committed Jun 2, 2020
1 parent de1f1af commit af14752
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions topi/include/topi/transform.h
Original file line number Diff line number Diff line change
Expand Up @@ -1019,11 +1019,11 @@ inline Tensor gather(const Tensor& data, int axis, const Tensor& indices,
}
Array<PrimExpr> real_indices;
for (size_t i = 0; i < ndim_i; ++i) {
if (i == (size_t)axis) {
real_indices.push_back(indices(indices_position));
} else {
real_indices.push_back(indices_position[i]);
}
if (i == (size_t)axis) {
real_indices.push_back(indices(indices_position));
} else {
real_indices.push_back(indices_position[i]);
}
}
return data(real_indices);
},
Expand Down

0 comments on commit af14752

Please sign in to comment.