Unverified Commit dadb1227 authored by yimeisun123's avatar yimeisun123 Committed by GitHub

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

parent ea40bc41
......@@ -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();
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment