test subgraph added nhwc format check

2022-05-13 19:07:23 +08:00
parent 901210fa6c
commit 543963bf8d
1 changed files with 2 additions and 1 deletions
--- a/src/tests/functional/plugin/cpu/subgraph_tests/src/subgraph_with_blocked_format.cpp
+++ b/src/tests/functional/plugin/cpu/subgraph_tests/src/subgraph_with_blocked_format.cpp
@@ -40,7 +40,8 @@ protected:
            if (layer_type == "Subgraph") {
                nodes_found++;
                auto output_layout = n->get_rt_info().at(ExecGraphInfoSerialization::OUTPUT_LAYOUTS).as<std::string>();
-                ASSERT_TRUE(output_layout == "aBcd8b" || output_layout == "aBcd16b");
+                // convolution maybe chooses 'nhwc' and the subgraph will follow it
+                ASSERT_TRUE(output_layout == "aBcd8b" || output_layout == "aBcd16b" || output_layout == "acdb");
            }
        }
        ASSERT_GT(nodes_found, 0);