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

Add test cases for PReLU in cpu plugin

Browse Source 
* For case when slope is vector
This commit is contained in:
Steve Yoo 2021-04-19 09:06:56 +09:00
parent db2de2fb19
commit 47dedfde57
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/activation.cpp
View File

@ -71,6 +71,10 @@ std::map<std::vector<size_t>, std::vector<std::vector<size_t>>> basic = {
std::map<std::vector<size_t>, std::vector<std::vector<size_t>>> 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(