Ilya Lavrenov
8f3d030d79
* Removed ov::runtime namespace * clang-format * Fixes * template reference * Fixes
74 lines
4.9 KiB
C++
74 lines
4.9 KiB
C++
// Copyright (C) 2018-2022 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "openvino/op/reduce_min.hpp"
|
|
#include "reduction.hpp"
|
|
|
|
using namespace ov;
|
|
using ReductionType = ngraph::helpers::ReductionType;
|
|
|
|
namespace reference_tests {
|
|
namespace ReductionOpsRefTestDefinitions {
|
|
namespace {
|
|
|
|
template <element::Type_t IN_ET>
|
|
std::vector<ReductionParams> generateReductionParams(const bool keep_dims) {
|
|
using T = typename element_type_traits<IN_ET>::value_type;
|
|
std::vector<ReductionParams> params = {
|
|
ReductionParams(ReductionType::Min, keep_dims, std::vector<int64_t>{0, 1},
|
|
reference_tests::Tensor({2, 2}, element::Type(IN_ET), std::vector<T>{1, 2, 3, 4}),
|
|
reference_tests::Tensor(ngraph::reduce(Shape{2, 2}, AxisSet{0, 1}, keep_dims), element::Type(IN_ET), std::vector<T>{1})),
|
|
ReductionParams(ReductionType::Min, keep_dims, std::vector<int64_t>{0},
|
|
reference_tests::Tensor({3, 2}, element::Type(IN_ET), std::vector<T>{1, 2, 3, 4, 5, 6}),
|
|
reference_tests::Tensor(ngraph::reduce(Shape{3, 2}, AxisSet{0}, keep_dims), element::Type(IN_ET), std::vector<T>{1, 2})),
|
|
ReductionParams(ReductionType::Min, keep_dims, std::vector<int64_t>{1},
|
|
reference_tests::Tensor({3, 2}, element::Type(IN_ET), std::vector<T>{1, 2, 3, 4, 5, 6}),
|
|
reference_tests::Tensor(ngraph::reduce(Shape{3, 2}, AxisSet{1}, keep_dims), element::Type(IN_ET), std::vector<T>{1, 3, 5})),
|
|
ReductionParams(ReductionType::Min, keep_dims, std::vector<int64_t>{0},
|
|
reference_tests::Tensor({3, 3, 3}, element::Type(IN_ET), std::vector<T>{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
|
|
16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27}),
|
|
reference_tests::Tensor(ngraph::reduce(Shape{3, 3, 3}, AxisSet{0}, keep_dims), element::Type(IN_ET), std::vector<T>{1, 2, 3, 4, 5, 6, 7, 8, 9})),
|
|
ReductionParams(ReductionType::Min, keep_dims, std::vector<int64_t>{2},
|
|
reference_tests::Tensor({3, 3, 3}, element::Type(IN_ET), std::vector<T>{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
|
|
16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27}),
|
|
reference_tests::Tensor(ngraph::reduce(Shape{3, 3, 3}, AxisSet{2}, keep_dims), element::Type(IN_ET), std::vector<T>{1, 4, 7, 10, 13, 16, 19, 22, 25})),
|
|
ReductionParams(ReductionType::Min, keep_dims, std::vector<int64_t>{0, 1},
|
|
reference_tests::Tensor({3, 3, 3}, element::Type(IN_ET), std::vector<T>{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
|
|
16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27}),
|
|
reference_tests::Tensor(ngraph::reduce(Shape{3, 3, 3}, AxisSet{0, 1}, keep_dims), element::Type(IN_ET), std::vector<T>{1, 2, 3})),
|
|
ReductionParams(ReductionType::Min, keep_dims, std::vector<int64_t>{0, 1, 2},
|
|
reference_tests::Tensor({3, 3, 3}, element::Type(IN_ET), std::vector<T>{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
|
|
16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27}),
|
|
reference_tests::Tensor(ngraph::reduce(Shape{3, 3, 3}, AxisSet{0, 1, 2}, keep_dims), element::Type(IN_ET), std::vector<T>{1}))
|
|
};
|
|
return params;
|
|
}
|
|
|
|
std::vector<ReductionParams> generateReductionCombinedParams() {
|
|
const std::vector<std::vector<ReductionParams>> reductionTypeParams {
|
|
generateReductionParams<element::Type_t::f32>(true),
|
|
generateReductionParams<element::Type_t::f32>(false),
|
|
generateReductionParams<element::Type_t::f16>(true),
|
|
generateReductionParams<element::Type_t::f16>(false),
|
|
generateReductionParams<element::Type_t::i32>(true),
|
|
generateReductionParams<element::Type_t::i32>(false),
|
|
generateReductionParams<element::Type_t::u32>(true),
|
|
generateReductionParams<element::Type_t::u32>(false),
|
|
generateReductionParams<element::Type_t::u64>(true),
|
|
generateReductionParams<element::Type_t::u64>(false)
|
|
};
|
|
std::vector<ReductionParams> combinedParams;
|
|
std::for_each(reductionTypeParams.begin(), reductionTypeParams.end(), [&](std::vector<ReductionParams> params) {
|
|
combinedParams.insert(combinedParams.end(), params.begin(), params.end());
|
|
});
|
|
return combinedParams;
|
|
}
|
|
} // namespace
|
|
INSTANTIATE_TEST_SUITE_P(smoke_Reduction_With_Hardcoded_Refs, ReferenceReductionLayerTest, ::testing::ValuesIn(generateReductionCombinedParams()),
|
|
ReferenceReductionLayerTest::getTestCaseName);
|
|
} // namespace ReductionOpsRefTestDefinitions
|
|
} // namespace reference_tests
|