Skip to content

N
ngraph
Commit 28e31004 authored by Amy Zhuang's avatar Amy Zhuang Committed by Scott Cyphers
Browse files
Options

Fix a bug in Gather Eigen kernel. (#3743)

parent 5fa63dda
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
src/ngraph/runtime/cpu/kernel/gather.hpp
View file @ 28e31004
......@@ -86,7 +86,6 @@ namespace ngraph
static_cast<ElementType*>(inputs), in_dims);
auto indices_ptr = static_cast<IndicesType*>(indices);
IndicesType index_value;
auto indices_rank = indices_shape.size();
auto outer_loop_num = 1;
for (int i = 0; i < axis; i++)
......@@ -126,7 +125,7 @@ namespace ngraph
// at axis
in_extents[axis] = 1;
// at axis, get the value from indices arg
index_value = indices_ptr[0];
IndicesType index_value = indices_ptr[0];
// take care of negative indices
in_offsets[axis] =
index_value >= 0 ? index_value : index_value + axis_length;
......@@ -199,7 +198,7 @@ namespace ngraph
}
// at axis, get the value from indices arg
int k = i % num_indices;
index_value = indices_ptr[k];
IndicesType index_value = indices_ptr[k];
// take care of negative indices
in_offsets[axis] =
index_value >= 0 ? index_value : index_value + axis_length;
......
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