[IE TESTS] Added single layer tests HSwish (#1861)
This commit is contained in:
@@ -47,7 +47,8 @@ const std::vector<ActivationTypes> activationTypes = {
|
||||
HardSigmoid,
|
||||
Selu,
|
||||
Ceiling,
|
||||
Mish
|
||||
Mish,
|
||||
HSwish
|
||||
};
|
||||
|
||||
const std::vector<ActivationTypes> activationParamTypes = {
|
||||
|
||||
@@ -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 = {
|
||||
|
||||
@@ -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<
|
||||
|
||||
@@ -108,7 +108,8 @@ enum ActivationTypes {
|
||||
Selu,
|
||||
Ceiling,
|
||||
PReLu,
|
||||
Mish
|
||||
Mish,
|
||||
HSwish
|
||||
};
|
||||
|
||||
enum EltwiseTypes {
|
||||
|
||||
@@ -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");
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user