[IE TESTS] Added single layer tests HSwish (#1861)

This commit is contained in:
Roman Lyamin
2020-08-20 13:32:16 +03:00
committed by GitHub
parent de8bfae243
commit 3a9e160031
5 changed files with 9 additions and 3 deletions

View File

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

View File

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

View File

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

View File

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