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

[VPU] - I64 issue with ONNX models - fix (#9978)

Browse Source 
i32 or i64 is used for index_element_type. So it is more convenient to get rid of the condition and stay only with the i32 option.
Tickets:
75748
75747
75029
This commit is contained in:
Daria Mityagina 2022-02-01 11:42:55 +03:00 committed by GitHub
parent 298cced3b3
commit a3f2a4ef99
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/plugins/intel_myriad/common/src/ngraph/transformations/dynamic_to_static_shape_topk.cpp
View File

@ -56,6 +56,7 @@ void dynamicToStaticShapeTopK(std::shared_ptr<ngraph::Node> target) {
} }
std::shared_ptr<ngraph::Node> new_topk; std::shared_ptr<ngraph::Node> new_topk;
if (ngraph::is_type<ngraph::opset3::Constant>(target->get_input_node_shared_ptr(1))) if (ngraph::is_type<ngraph::opset3::Constant>(target->get_input_node_shared_ptr(1)))
new_topk = target->clone_with_new_inputs(target->input_values()); new_topk = target->clone_with_new_inputs(target->input_values());
else else
@ -65,7 +66,7 @@ void dynamicToStaticShapeTopK(std::shared_ptr<ngraph::Node> target) {
topk->get_provided_axis(), topk->get_provided_axis(),
topk->get_mode(), topk->get_mode(),
topk->get_sort_type(), topk->get_sort_type(),
topk->get_index_element_type()); ngraph::element::i32);
for (auto &output : target->outputs()) { for (auto &output : target->outputs()) {
const auto outDSR = std::make_shared<ngraph::vpu::op::DynamicShapeResolver>(new_topk->output(output.get_index()), output_shape); const auto outDSR = std::make_shared<ngraph::vpu::op::DynamicShapeResolver>(new_topk->output(output.get_index()), output_shape);