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

TopK base class cleanup (#16154)

Browse Source
This commit is contained in:
Tomasz Dołbniak authored and GitHub committed 2023-03-08 11:33:38 +00:00
1 parent 50b76873e2
commit e5ef0fee8e
6 files changed
+65 -70
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/frontends/onnx/tests/onnx_tensor_names.cpp
+2 -2
View File
@@ -69,8 +69,8 @@ NGRAPH_TEST(onnx_tensor_names, node_multiple_outputs) {
const auto ops = function->get_ordered_ops();
EXPECT_TRUE(matching_node_found_in_graph<op::Parameter>(ops, "x", {"x"}));
EXPECT_TRUE(matching_node_found_in_graph<ov::op::util::TopK_Base>(ops, "indices", {"values"}, 0));
EXPECT_TRUE(matching_node_found_in_graph<ov::op::util::TopK_Base>(ops, "indices", {"indices"}, 1));
EXPECT_TRUE(matching_node_found_in_graph<ov::op::util::TopKBase>(ops, "indices", {"values"}, 0));
EXPECT_TRUE(matching_node_found_in_graph<ov::op::util::TopKBase>(ops, "indices", {"indices"}, 1));
const auto results = function->get_results();
EXPECT_TRUE(matching_node_found_in_graph<op::Result>(results, "values/sink_port_0", {"values"}));