Unverified Commit 35d50450 authored by Evgenya Stepyreva's avatar Evgenya Stepyreva Committed by GitHub

FQ output data type copied from 3,4 input data type (#4341)

* FQ output data type copied from 3,4 input data type

* style
Co-authored-by: 's avatarScott Cyphers <diyessi@users.noreply.github.com>
parent 88bcb685
......@@ -77,7 +77,18 @@ void op::FakeQuantize::validate_and_infer_types()
NODE_VALIDATION_CHECK(this, false, "Unsupported auto broadcast specification");
}
}
set_output_type(0, get_input_element_type(0), get_input_partial_shape(0));
element::Type result_et;
NODE_VALIDATION_CHECK(
this,
element::Type::merge(result_et, get_input_element_type(3), get_input_element_type(4)),
"Arguments do not have the same element type (arg3 element type: ",
get_input_element_type(3),
", arg4 element type: ",
get_input_element_type(4),
").");
set_output_type(0, result_et, get_input_partial_shape(0));
}
NodeVector op::FakeQuantize::decompose_op() const
......
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