IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add default infer_probs (#14653)

Browse Source 
* add default infer_probs

* fix clang format

---------

Co-authored-by: Ilya Churaev <ilya.churaev@intel.com>
This commit is contained in:
Wilson Seok 2023-02-08 14:27:13 +09:00 committed by GitHub
parent 7b649c4150
commit 18905ada20
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/core/src/op/proposal.cpp
View File

@ -98,6 +98,7 @@ void op::v4::Proposal::validate_and_infer_types() {
const auto& input0_type = get_input_element_type(0); const auto& input0_type = get_input_element_type(0);
set_output_type(0, input0_type, output_shapes[0]); set_output_type(0, input0_type, output_shapes[0]);
set_output_type(1, input0_type, output_shapes[1]); 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<Node> op::v4::Proposal::clone_with_new_inputs(const OutputVector& new_args) const { std::shared_ptr<Node> op::v4::Proposal::clone_with_new_inputs(const OutputVector& new_args) const {