[IE TESTS] Added single layer test Mish (#1401)

This commit is contained in:
Roman Lyamin 2020-07-21 15:11:05 +03:00 committed by GitHub
parent eecd03aa85
commit 613d822458
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 10 additions and 2 deletions

View File

@ -47,6 +47,7 @@ const std::vector<ActivationTypes> activationTypes = {
HardSigmoid, HardSigmoid,
Selu, Selu,
Ceiling, Ceiling,
Mish
}; };
const std::vector<ActivationTypes> activationParamTypes = { const std::vector<ActivationTypes> activationParamTypes = {

View File

@ -51,5 +51,7 @@ std::vector<std::string> disabledTestPatterns() {
R"(.*ActivationLayerTest.*Ceiling.*)", R"(.*ActivationLayerTest.*Ceiling.*)",
// TODO: Issue: 32032 // TODO: Issue: 32032
R"(.*ActivationParamLayerTest.*)", R"(.*ActivationParamLayerTest.*)",
// TODO: Issue: 32959
R"(.*ActivationLayerTest.*Mish.*)",
}; };
} }

View File

@ -41,7 +41,8 @@ const std::vector<ActivationTypes> activationTypes = {
Erf, Erf,
HardSigmoid, HardSigmoid,
Selu, Selu,
Ceiling Ceiling,
Mish
}; };
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>>> basic = {

View File

@ -68,6 +68,7 @@ static std::map<ngraph::helpers::ActivationTypes, std::string> activationNames =
{ngraph::helpers::ActivationTypes::Gelu, "Gelu"}, {ngraph::helpers::ActivationTypes::Gelu, "Gelu"},
{ngraph::helpers::ActivationTypes::Ceiling, "Ceiling"}, {ngraph::helpers::ActivationTypes::Ceiling, "Ceiling"},
{ngraph::helpers::ActivationTypes::PReLu, "PReLu"}, {ngraph::helpers::ActivationTypes::PReLu, "PReLu"},
{ngraph::helpers::ActivationTypes::Mish, "Mish"},
}; };
typedef std::tuple< typedef std::tuple<

View File

@ -107,7 +107,8 @@ enum ActivationTypes {
HardSigmoid, HardSigmoid,
Selu, Selu,
Ceiling, Ceiling,
PReLu PReLu,
Mish
}; };
enum EltwiseTypes { enum EltwiseTypes {

View File

@ -91,6 +91,8 @@ std::shared_ptr<ngraph::Node> makeActivation(const ngraph::Output<Node> &in,
std::vector<float>{-0.01f}); std::vector<float>{-0.01f});
return std::make_shared<ngraph::op::PRelu>(in, negative_slope); return std::make_shared<ngraph::op::PRelu>(in, negative_slope);
} }
case ngraph::helpers::ActivationTypes::Mish:
return std::make_shared<ngraph::op::v4::Mish>(in);
default: default:
throw std::runtime_error("Can't create layer for this activation type"); throw std::runtime_error("Can't create layer for this activation type");
} }