[IE TESTS] Added single layer test Mish (#1401)
This commit is contained in:
parent
eecd03aa85
commit
613d822458
@ -47,6 +47,7 @@ const std::vector<ActivationTypes> activationTypes = {
|
||||
HardSigmoid,
|
||||
Selu,
|
||||
Ceiling,
|
||||
Mish
|
||||
};
|
||||
|
||||
const std::vector<ActivationTypes> activationParamTypes = {
|
||||
|
@ -51,5 +51,7 @@ std::vector<std::string> disabledTestPatterns() {
|
||||
R"(.*ActivationLayerTest.*Ceiling.*)",
|
||||
// TODO: Issue: 32032
|
||||
R"(.*ActivationParamLayerTest.*)",
|
||||
// TODO: Issue: 32959
|
||||
R"(.*ActivationLayerTest.*Mish.*)",
|
||||
};
|
||||
}
|
@ -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 = {
|
||||
|
@ -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<
|
||||
|
@ -107,7 +107,8 @@ enum ActivationTypes {
|
||||
HardSigmoid,
|
||||
Selu,
|
||||
Ceiling,
|
||||
PReLu
|
||||
PReLu,
|
||||
Mish
|
||||
};
|
||||
|
||||
enum EltwiseTypes {
|
||||
|
@ -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");
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user