82 lines
2.7 KiB
C++
82 lines
2.7 KiB
C++
// Copyright (C) 2018-2021 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#include <vector>
|
|
|
|
#include "shared_test_classes/single_layer/eltwise.hpp"
|
|
|
|
using namespace ov::test::subgraph;
|
|
|
|
namespace {
|
|
TEST_P(EltwiseLayerTest, Serialize) {
|
|
serialize();
|
|
}
|
|
|
|
const std::vector<ov::test::ElementType> inputPrecisions = {
|
|
ov::element::f32,
|
|
ov::element::f16,
|
|
ov::element::i32,
|
|
};
|
|
|
|
std::vector<std::vector<ov::Shape>> inputShapes = {
|
|
{{2}},
|
|
{{1, 5, 50}},
|
|
{{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 = {
|
|
ngraph::helpers::InputLayerType::CONSTANT,
|
|
ngraph::helpers::InputLayerType::PARAMETER,
|
|
};
|
|
|
|
std::vector<CommonTestUtils::OpType> opTypes = {
|
|
CommonTestUtils::OpType::SCALAR,
|
|
CommonTestUtils::OpType::VECTOR,
|
|
};
|
|
|
|
std::vector<ngraph::helpers::EltwiseTypes> eltwiseOpTypes = {
|
|
ngraph::helpers::EltwiseTypes::ADD,
|
|
ngraph::helpers::EltwiseTypes::MULTIPLY,
|
|
ngraph::helpers::EltwiseTypes::SUBTRACT,
|
|
ngraph::helpers::EltwiseTypes::DIVIDE,
|
|
ngraph::helpers::EltwiseTypes::FLOOR_MOD,
|
|
ngraph::helpers::EltwiseTypes::SQUARED_DIFF,
|
|
ngraph::helpers::EltwiseTypes::POWER,
|
|
ngraph::helpers::EltwiseTypes::MOD
|
|
};
|
|
|
|
std::map<std::string, std::string> additionalConfig = {};
|
|
|
|
const auto elementiwiseParams = ::testing::Combine(
|
|
::testing::ValuesIn(ov::test::static_shapes_to_test_representation(inputShapes)),
|
|
::testing::ValuesIn(eltwiseOpTypes),
|
|
::testing::ValuesIn(secondaryInputTypes),
|
|
::testing::ValuesIn(opTypes),
|
|
::testing::ValuesIn(inputPrecisions),
|
|
::testing::Values(CommonTestUtils::DEVICE_CPU),
|
|
::testing::Values(additionalConfig));
|
|
|
|
const auto elementiwiseParamsDyn = ::testing::Combine(
|
|
::testing::ValuesIn(inShapesDynamic),
|
|
::testing::ValuesIn(eltwiseOpTypes),
|
|
::testing::ValuesIn(secondaryInputTypes),
|
|
::testing::ValuesIn(opTypes),
|
|
::testing::ValuesIn(inputPrecisions),
|
|
::testing::Values(CommonTestUtils::DEVICE_CPU),
|
|
::testing::Values(additionalConfig));
|
|
|
|
INSTANTIATE_TEST_SUITE_P(smoke_ElementwiseSerialization_static, EltwiseLayerTest,
|
|
elementiwiseParams,
|
|
EltwiseLayerTest::getTestCaseName);
|
|
|
|
INSTANTIATE_TEST_SUITE_P(smoke_ElementwiseSerialization_dynamic, EltwiseLayerTest,
|
|
elementiwiseParamsDyn,
|
|
EltwiseLayerTest::getTestCaseName);
|
|
} // namespace
|