From 18905ada20005e8de320ad7661f4bb9f82be91f7 Mon Sep 17 00:00:00 2001 From: Wilson Seok Date: Wed, 8 Feb 2023 14:27:13 +0900 Subject: [PATCH] add default infer_probs (#14653) * add default infer_probs * fix clang format --------- Co-authored-by: Ilya Churaev --- src/core/src/op/proposal.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/src/core/src/op/proposal.cpp b/src/core/src/op/proposal.cpp index 96d094303fc..18941d3e014 100644 --- a/src/core/src/op/proposal.cpp +++ b/src/core/src/op/proposal.cpp @@ -98,6 +98,7 @@ void op::v4::Proposal::validate_and_infer_types() { const auto& input0_type = get_input_element_type(0); set_output_type(0, input0_type, output_shapes[0]); set_output_type(1, input0_type, output_shapes[1]); + m_attrs.infer_probs = true; // Proposal v4 requires default true of infer_probs so output_1 has valid data. } std::shared_ptr op::v4::Proposal::clone_with_new_inputs(const OutputVector& new_args) const {