[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,
Selu,
Ceiling,
Mish
};
const std::vector<ActivationTypes> activationParamTypes = {

View File

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

View File

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

View File

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

View File

@ -91,6 +91,8 @@ std::shared_ptr<ngraph::Node> makeActivation(const ngraph::Output<Node> &in,
std::vector<float>{-0.01f});
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:
throw std::runtime_error("Can't create layer for this activation type");
}