diff --git a/src/plugins/intel_myriad/common/src/ngraph/transformations/dynamic_to_static_shape_topk.cpp b/src/plugins/intel_myriad/common/src/ngraph/transformations/dynamic_to_static_shape_topk.cpp index 62bcae6ac13..67df532d619 100644 --- a/src/plugins/intel_myriad/common/src/ngraph/transformations/dynamic_to_static_shape_topk.cpp +++ b/src/plugins/intel_myriad/common/src/ngraph/transformations/dynamic_to_static_shape_topk.cpp @@ -56,6 +56,7 @@ void dynamicToStaticShapeTopK(std::shared_ptr target) { } std::shared_ptr new_topk; + if (ngraph::is_type(target->get_input_node_shared_ptr(1))) new_topk = target->clone_with_new_inputs(target->input_values()); else @@ -65,7 +66,7 @@ void dynamicToStaticShapeTopK(std::shared_ptr target) { topk->get_provided_axis(), topk->get_mode(), topk->get_sort_type(), - topk->get_index_element_type()); + ngraph::element::i32); for (auto &output : target->outputs()) { const auto outDSR = std::make_shared(new_topk->output(output.get_index()), output_shape);