Commit 00fc1a25 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format.

parent 04e01f74
...@@ -653,7 +653,10 @@ struct gather ...@@ -653,7 +653,10 @@ struct gather
} }
template <class T> template <class T>
void compute_index(const T& out_idx, const std::vector<std::size_t>& vec_indices, const std::size_t max_dim, T& in_idx) const void compute_index(const T& out_idx,
const std::vector<std::size_t>& vec_indices,
const std::size_t max_dim,
T& in_idx) const
{ {
in_idx = out_idx; in_idx = out_idx;
std::size_t idx = vec_indices.at(out_idx[axis]); std::size_t idx = vec_indices.at(out_idx[axis]);
......
...@@ -219,7 +219,9 @@ TEST_CASE(gather) ...@@ -219,7 +219,9 @@ TEST_CASE(gather)
migraphx::shape indices{migraphx::shape::int32_type, {2, 3}}; migraphx::shape indices{migraphx::shape::int32_type, {2, 3}};
std::size_t axis = 1; std::size_t axis = 1;
expect_shape(migraphx::shape{migraphx::shape::float_type, {2, 6, 4, 5}}, expect_shape(migraphx::shape{migraphx::shape::float_type, {2, 6, 4, 5}},
migraphx::op::gather{axis}, input, indices); migraphx::op::gather{axis},
input,
indices);
} }
{ {
......
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