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, ...@@ -52,7 +52,7 @@ op::FakeQuantize::FakeQuantize(const Output<Node>& data,
constructor_validate_and_infer_types(); 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); PartialShape data_pshape = get_input_partial_shape(0);
...@@ -77,6 +77,7 @@ void op::FakeQuantize::pre_validate_and_infer_types() ...@@ -77,6 +77,7 @@ void op::FakeQuantize::pre_validate_and_infer_types()
NODE_VALIDATION_CHECK(this, false, "Unsupported auto broadcast specification"); 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 NodeVector op::FakeQuantize::decompose_op() const
......
...@@ -68,7 +68,7 @@ namespace ngraph ...@@ -68,7 +68,7 @@ namespace ngraph
AutoBroadcastSpec(AutoBroadcastType::NUMPY)); AutoBroadcastSpec(AutoBroadcastType::NUMPY));
virtual NodeVector decompose_op() const override; 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> virtual std::shared_ptr<Node>
copy_with_new_args(const NodeVector& new_args) const override; 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