From 47dedfde5780480ea036fe5eadae3233a68c3c96 Mon Sep 17 00:00:00 2001 From: Steve Yoo Date: Mon, 19 Apr 2021 09:06:56 +0900 Subject: [PATCH] Add test cases for PReLU in cpu plugin * For case when slope is vector --- .../shared_tests_instances/single_layer_tests/activation.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/activation.cpp b/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/activation.cpp index 5decd0d9c88..dfcee6bd5fa 100644 --- a/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/activation.cpp +++ b/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/activation.cpp @@ -71,6 +71,10 @@ std::map, std::vector>> basic = { std::map, std::vector>> preluBasic = { {{1, 50}, {{1}, {50}}}, {{1, 128}, {{1}, {128}}}, + {{20, 128}, {{20}, {128}, {20, 128}}}, + {{1, 20, 128}, {{1}, {20}, {128}, {20, 128}}}, + {{1, 20, 128, 128}, {{1}, {20}, {128}, {128, 128}, {20, 128, 128}}}, + {{1, 20, 20, 128, 128}, {{1}, {20}, {128}, {128, 128}, {20, 128, 128}, {20, 20, 128, 128}}}, }; const auto basicCases = ::testing::Combine(