Fix get_index_element_type to return the correct type for IndexReduct…

…ion OP (#1709)
NervanaSystems · Sep 27, 2018 · dadb122 · dadb122
1 parent ea40bc4
commit dadb122
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/ngraph/op/util/index_reduction.cpp b/src/ngraph/op/util/index_reduction.cpp
@@ -27,6 +27,7 @@ op::util::IndexReduction::IndexReduction(const std::string& node_type,
                                          const element::Type& index_element_type)
     : Op(node_type, check_single_output_args({arg}))
     , m_axis(axis)
+    , m_index_element_type(index_element_type)
 {
     constructor_validate_and_infer_types();
     auto rank = arg->get_shape().size();