Fix get_index_element_type to return the correct type for IndexReduction OP (#1709)

yimeisun123 · web-flow · commit dadb12270e7f · 2018-09-27T15:31:43.000-07:00
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();

Original file line number	Diff line number	Diff line change
`@@ -27,6 +27,7 @@ op::util::IndexReduction::IndexReduction(const std::string& node_type,`
`27`	`27`	`const element::Type& index_element_type)`
`28`	`28`	`: Op(node_type, check_single_output_args({arg}))`
`29`	`29`	`, m_axis(axis)`
	`30`	`+ , m_index_element_type(index_element_type)`
`30`	`31`	`{`
`31`	`32`	`constructor_validate_and_infer_types();`
`32`	`33`	`auto rank = arg->get_shape().size();`