fix oss build

This commit is contained in:
Efode, Irina 2021-10-14 17:05:24 +03:00
parent 537c0536f9
commit 2b19f6c607

View File

@ -6,23 +6,28 @@
#include "shared_test_classes/single_layer/eltwise.hpp" #include "shared_test_classes/single_layer/eltwise.hpp"
using namespace LayerTestsDefinitions; using namespace ov::test::subgraph;
namespace { namespace {
TEST_P(EltwiseLayerTest, Serialize) { TEST_P(EltwiseLayerTest, Serialize) {
Serialize(); serialize();
} }
const std::vector<InferenceEngine::Precision> inputPrecisions = { const std::vector<ov::test::ElementType> inputPrecisions = {
InferenceEngine::Precision::FP32, ov::element::f32,
InferenceEngine::Precision::FP16, ov::element::f16,
InferenceEngine::Precision::I32, ov::element::i32,
}; };
std::vector<std::pair<std::vector<ngraph::PartialShape>, std::vector<std::vector<ngraph::Shape>>>> inputShapes = { std::vector<std::vector<ov::Shape>> inputShapes = {
{{}, {{{2}}}}, {{2}},
{{}, {{{1, 5, 50}}}}, {{1, 5, 50}},
{{}, {{{2, 10, 1, 4}, {2, 10, 1, 1}}}} {{2, 10, 1, 4}, {2, 10, 1, 1}}
};
std::vector<ov::test::InputShapes> inShapesDynamic = {
{{{ngraph::Dimension(1, 10), 200}, {ngraph::Dimension(1, 10), 200}},
{{{2, 200}, {2, 200}}, {{1, 200}, {5, 200}}}},
}; };
std::vector<ngraph::helpers::InputLayerType> secondaryInputTypes = { std::vector<ngraph::helpers::InputLayerType> secondaryInputTypes = {
@ -49,18 +54,32 @@ std::vector<ngraph::helpers::EltwiseTypes> eltwiseOpTypes = {
std::map<std::string, std::string> additionalConfig = {}; std::map<std::string, std::string> additionalConfig = {};
const auto elementiwiseParams = ::testing::Combine( const auto elementiwiseParams = ::testing::Combine(
::testing::ValuesIn(inputShapes), ::testing::ValuesIn(ov::test::static_shapes_to_test_representation(inputShapes)),
::testing::ValuesIn(eltwiseOpTypes), ::testing::ValuesIn(eltwiseOpTypes),
::testing::ValuesIn(secondaryInputTypes), ::testing::ValuesIn(secondaryInputTypes),
::testing::ValuesIn(opTypes), ::testing::ValuesIn(opTypes),
::testing::ValuesIn(inputPrecisions), ::testing::ValuesIn(inputPrecisions),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED), ::testing::Values(ov::element::undefined),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED), ::testing::Values(ov::element::undefined),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(CommonTestUtils::DEVICE_CPU), ::testing::Values(CommonTestUtils::DEVICE_CPU),
::testing::Values(additionalConfig)); ::testing::Values(additionalConfig));
INSTANTIATE_TEST_SUITE_P(smoke_ElementwiseSerialization, EltwiseLayerTest, const auto elementiwiseParamsDyn = ::testing::Combine(
::testing::ValuesIn(inShapesDynamic),
::testing::ValuesIn(eltwiseOpTypes),
::testing::ValuesIn(secondaryInputTypes),
::testing::ValuesIn(opTypes),
::testing::ValuesIn(inputPrecisions),
::testing::Values(ov::element::undefined),
::testing::Values(ov::element::undefined),
::testing::Values(CommonTestUtils::DEVICE_CPU),
::testing::Values(additionalConfig));
INSTANTIATE_TEST_SUITE_P(smoke_ElementwiseSerialization_static, EltwiseLayerTest,
elementiwiseParams, elementiwiseParams,
EltwiseLayerTest::getTestCaseName); EltwiseLayerTest::getTestCaseName);
INSTANTIATE_TEST_SUITE_P(smoke_ElementwiseSerialization_dynamic, EltwiseLayerTest,
elementiwiseParamsDyn,
EltwiseLayerTest::getTestCaseName);
} // namespace } // namespace