From dadb12270e7fef53ec506a826fb258e22af59ea5 Mon Sep 17 00:00:00 2001 From: yimeisun123 <42156420+yimeisun123@users.noreply.github.com> Date: Thu, 27 Sep 2018 15:31:43 -0700 Subject: [PATCH] Fix get_index_element_type to return the correct type for IndexReduction OP (#1709) --- src/ngraph/op/util/index_reduction.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/src/ngraph/op/util/index_reduction.cpp b/src/ngraph/op/util/index_reduction.cpp index 9c455c514b6..da0c64abb94 100644 --- 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();