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

[CPU][IE TESTS] Covered fusing of HSwish, Mish to Conv by tests (#4903)

Browse Source
This commit is contained in:
Alexandra Sidorova 2021-03-26 10:48:54 +03:00 committed by GitHub
parent fe26fc26a7
commit 0cfd007206
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference-engine/tests/functional/plugin/cpu/single_layer_tests/convolution.cpp
View File

@ -110,6 +110,8 @@ const std::vector<fusingSpecificParams> fusingParamsSet{
fusingClamp,
fusingPRelu,
fusingSwish,
fusingHSwish,
fusingMish,
// other patterns
fusingReluScaleShift,
fusingFakeQuantizePerTensorRelu,

2
inference-engine/tests/functional/plugin/cpu/single_layer_tests/group_convolution.cpp
View File

@ -121,6 +121,8 @@ std::vector<fusingSpecificParams> fusingParamsSet {
fusingClamp,
fusingPRelu,
fusingSwish,
fusingHSwish,
fusingMish,
// other patterns
fusingReluScaleShift,
fusingFakeQuantizePerTensorRelu,

8
inference-engine/tests/functional/plugin/cpu/test_utils/fusing_test_utils.hpp
View File

@ -104,6 +104,14 @@ const auto fusingSwish = fusingSpecificParams{std::make_shared<postNodesMgr>(std
{[](std::shared_ptr<ngraph::Node> inpNode, const ngraph::element::Type& ngPrc, ngraph::ParameterVector& params){
return ngraph::builder::makeActivation(inpNode, ngPrc, ngraph::helpers::Swish, {}, {1.0f});
}, "Swish"}}), {"Swish"}};
const auto fusingHSwish = fusingSpecificParams{std::make_shared<postNodesMgr>(std::vector<postNodeBuilder>{
{[](std::shared_ptr<ngraph::Node> inpNode, const ngraph::element::Type& ngPrc, ngraph::ParameterVector& params){
return ngraph::builder::makeActivation(inpNode, ngPrc, ngraph::helpers::HSwish, {}, {});
}, "HSwish"}}), {"HSwish"}};
const auto fusingMish = fusingSpecificParams{std::make_shared<postNodesMgr>(std::vector<postNodeBuilder>{
{[](std::shared_ptr<ngraph::Node> inpNode, const ngraph::element::Type& ngPrc, ngraph::ParameterVector& params){
return ngraph::builder::makeActivation(inpNode, ngPrc, ngraph::helpers::Mish, {}, {});
}, "Mish"}}), {"Mish"}};
const auto fusingReluScaleShift = fusingSpecificParams{std::make_shared<postNodesMgr>(std::vector<postNodeBuilder>{
{[](std::shared_ptr<ngraph::Node> inpNode, const ngraph::element::Type& ngPrc, ngraph::ParameterVector& params){