Fixed Proposal output type infer (#3538)

0c0b5105 · GlebKazantaev · Scott Cyphers · b18cb73d · 0c0b5105
Commit 0c0b5105 authored Aug 30, 2019 by GlebKazantaev Committed by Scott Cyphers Aug 30, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

proposal.cpp src/ngraph/op/experimental/layers/proposal.cpp +2 -2

No files found.
--- a/src/ngraph/op/experimental/layers/proposal.cpp
+++ b/src/ngraph/op/experimental/layers/proposal.cpp
@@ -76,11 +76,11 @@ void op::Proposal::validate_and_infer_types()
            ").");

        auto batch_size = class_probs_shape[0];
-        set_output_type(0, element::f32, Shape{batch_size * m_attrs.post_nms_topn, 5});
+        set_output_type(0, get_input_element_type(0), Shape{batch_size * m_attrs.post_nms_topn, 5});
    }
    else
    {
-        set_output_type(0, element::f32, PartialShape::dynamic());
+        set_output_type(0, get_input_element_type(0), PartialShape::dynamic());
    }
 }