Commit 06e02c28 authored by Ilya Churaev's avatar Ilya Churaev Committed by Scott Cyphers

Do direct FakeQuantize type propagation (#4220)

Co-authored-by: 's avatarScott Cyphers <diyessi@users.noreply.github.com>
parent 9cc2ce34
......@@ -52,7 +52,7 @@ op::FakeQuantize::FakeQuantize(const Output<Node>& data,
constructor_validate_and_infer_types();
}
void op::FakeQuantize::pre_validate_and_infer_types()
void op::FakeQuantize::validate_and_infer_types()
{
PartialShape data_pshape = get_input_partial_shape(0);
......@@ -77,6 +77,7 @@ void op::FakeQuantize::pre_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));
}
NodeVector op::FakeQuantize::decompose_op() const
......
......@@ -68,7 +68,7 @@ namespace ngraph
AutoBroadcastSpec(AutoBroadcastType::NUMPY));
virtual NodeVector decompose_op() const override;
virtual void pre_validate_and_infer_types() override;
virtual void validate_and_infer_types() override;
virtual std::shared_ptr<Node>
copy_with_new_args(const NodeVector& new_args) const override;
......
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