[ONNX] Add support for negative axis attribute in Gather op. (#2902)

5ec02d8e · tsocha · Scott Cyphers · a006b5c4 · 5ec02d8e
Commit 5ec02d8e authored May 10, 2019 by tsocha Committed by Scott Cyphers May 10, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

gather.hpp src/ngraph/frontend/onnx_import/op/gather.hpp +4 -0

No files found.
--- a/src/ngraph/frontend/onnx_import/op/gather.hpp
+++ b/src/ngraph/frontend/onnx_import/op/gather.hpp
@@ -34,6 +34,10 @@ namespace ngraph
                    auto data = ng_inputs.at(0);
                    auto indices = ng_inputs.at(1);
                    auto axis = node.get_attribute_value<int64_t>("axis", 0);
+                    if (axis < 0)
+                    {
+                        axis += data->get_shape().size();
+                    }

                    return {std::make_shared<ngraph::op::Gather>(data, indices, axis)};
                }