[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,
|
HardSigmoid,
|
||||||
Selu,
|
Selu,
|
||||||
Ceiling,
|
Ceiling,
|
||||||
|
Mish
|
||||||
};
|
};
|
||||||
|
|
||||||
const std::vector<ActivationTypes> activationParamTypes = {
|
const std::vector<ActivationTypes> activationParamTypes = {
|
||||||
|
@ -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.*)",
|
||||||
};
|
};
|
||||||
}
|
}
|
@ -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 = {
|
||||||
|
@ -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<
|
||||||
|
@ -107,7 +107,8 @@ enum ActivationTypes {
|
|||||||
HardSigmoid,
|
HardSigmoid,
|
||||||
Selu,
|
Selu,
|
||||||
Ceiling,
|
Ceiling,
|
||||||
PReLu
|
PReLu,
|
||||||
|
Mish
|
||||||
};
|
};
|
||||||
|
|
||||||
enum EltwiseTypes {
|
enum EltwiseTypes {
|
||||||
|
@ -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");
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user