Merge branch 'gather_operator' into seq2seq_example

077970f7 · Shucai Xiao · 9c16a90e · 463026f9 · 077970f7
Commit 077970f7 authored Feb 25, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

src/include/migraphx/operators.hpp src/include/migraphx/operators.hpp +4 -2

No files found.
--- a/src/include/migraphx/operators.hpp
+++ b/src/include/migraphx/operators.hpp
@@ -774,8 +774,10 @@ struct gather
        return {type, lens};
    }

-    template <typename V, typename T>
-    T compute_data_index(const V& indices, const int axis_index, const T& out_idx) const
+    template <typename V>
+    std::vector<std::size_t> compute_data_index(const V& indices,
+                                                const int axis_index,
+                                                const std::vector<std::size_t>& out_idx) const
    {
        auto data_idx = out_idx;
        std::size_t index{};