Migrate Movement 1 (Gather-1, Gather-7, Pad-1) (#8414)
This commit is contained in:
Steve Yoo
GitHub
parent
beaec484bb
commit
b320061ea3
880
docs/template_plugin/tests/functional/op_reference/gather.cpp
Normal file
880
docs/template_plugin/tests/functional/op_reference/gather.cpp
Normal file
@ -0,0 +1,880 @@
|
||||
// Copyright (C) 2021 Intel Corporation
|
||||
// SPDX-License-Identifier: Apache-2.0
|
||||
//
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
#include "openvino/op/gather.hpp"
|
||||
#include "openvino/op/constant.hpp"
|
||||
#include "base_reference_test.hpp"
|
||||
|
||||
using namespace reference_tests;
|
||||
using namespace ov;
|
||||
|
||||
namespace {
|
||||
struct GatherParams {
|
||||
GatherParams(
|
||||
const Tensor& dataTensor, const Tensor& indicesTensor, const Tensor& axisTensor,
|
||||
const Tensor& expectedTensor, const std::string& testcaseName = "") :
|
||||
dataTensor(dataTensor), indicesTensor(indicesTensor), axisTensor(axisTensor),
|
||||
expectedTensor(expectedTensor), testcaseName(testcaseName) {}
|
||||
|
||||
Tensor dataTensor;
|
||||
Tensor indicesTensor;
|
||||
Tensor axisTensor;
|
||||
Tensor expectedTensor;
|
||||
std::string testcaseName;
|
||||
};
|
||||
|
||||
class ReferenceGatherTest : public testing::TestWithParam<GatherParams>, public CommonReferenceTest {
|
||||
public:
|
||||
void SetUp() override {
|
||||
auto params = GetParam();
|
||||
function = CreateFunction(params);
|
||||
inputData = {params.dataTensor.data, params.indicesTensor.data};
|
||||
refOutData = {params.expectedTensor.data};
|
||||
}
|
||||
|
||||
static std::string getTestCaseName(const testing::TestParamInfo<GatherParams>& obj) {
|
||||
auto param = obj.param;
|
||||
std::ostringstream result;
|
||||
result << "dType=" << param.dataTensor.type;
|
||||
result << "_dShape=" << param.dataTensor.shape;
|
||||
result << "_iType=" << param.indicesTensor.type;
|
||||
result << "_iShape=" << param.indicesTensor.shape;
|
||||
result << "_aType=" << param.axisTensor.type;
|
||||
result << "_aShape=" << param.axisTensor.shape;
|
||||
result << "_eType=" << param.expectedTensor.type;
|
||||
if (param.testcaseName != "") {
|
||||
result << "_eShape=" << param.expectedTensor.shape;
|
||||
result << "_=" << param.testcaseName;
|
||||
} else {
|
||||
result << "_eShape=" << param.expectedTensor.shape;
|
||||
}
|
||||
return result.str();
|
||||
}
|
||||
|
||||
private:
|
||||
static std::shared_ptr<Function> CreateFunction(const GatherParams& params) {
|
||||
const auto P = std::make_shared<op::v0::Parameter>(params.dataTensor.type, params.dataTensor.shape);
|
||||
const auto I = std::make_shared<op::v0::Parameter>(params.indicesTensor.type, params.indicesTensor.shape);
|
||||
const auto A = op::v0::Constant::create(params.axisTensor.type, params.axisTensor.shape,
|
||||
params.axisTensor.data.data());
|
||||
const auto G = std::make_shared<op::v1::Gather>(P, I, A);
|
||||
const auto f = std::make_shared<Function>(G, ParameterVector{P, I});
|
||||
return f;
|
||||
}
|
||||
};
|
||||
|
||||
TEST_P(ReferenceGatherTest, CompareWithRefs) {
|
||||
Exec();
|
||||
}
|
||||
|
||||
struct GatherParamsV7 {
|
||||
GatherParamsV7(
|
||||
const Tensor& dataTensor, const Tensor& indicesTensor, const Tensor& axisTensor, int64_t batchDims,
|
||||
const Tensor& expectedTensor, const std::string& testcaseName = "") :
|
||||
dataTensor(dataTensor), indicesTensor(indicesTensor), axisTensor(axisTensor), batchDims(batchDims),
|
||||
expectedTensor(expectedTensor), testcaseName(testcaseName) {}
|
||||
|
||||
Tensor dataTensor;
|
||||
Tensor indicesTensor;
|
||||
Tensor axisTensor;
|
||||
int64_t batchDims;
|
||||
Tensor expectedTensor;
|
||||
std::string testcaseName;
|
||||
};
|
||||
|
||||
class ReferenceGatherTestV7 : public testing::TestWithParam<GatherParamsV7>, public CommonReferenceTest {
|
||||
public:
|
||||
void SetUp() override {
|
||||
auto params = GetParam();
|
||||
function = CreateFunction(params);
|
||||
inputData = {params.dataTensor.data, params.indicesTensor.data};
|
||||
refOutData = {params.expectedTensor.data};
|
||||
}
|
||||
|
||||
static std::string getTestCaseName(const testing::TestParamInfo<GatherParamsV7>& obj) {
|
||||
auto param = obj.param;
|
||||
std::ostringstream result;
|
||||
result << "dType=" << param.dataTensor.type;
|
||||
result << "_dShape=" << param.dataTensor.shape;
|
||||
result << "_iType=" << param.indicesTensor.type;
|
||||
result << "_iShape=" << param.indicesTensor.shape;
|
||||
result << "_aType=" << param.axisTensor.type;
|
||||
result << "_aShape=" << param.axisTensor.shape;
|
||||
result << "_batchDims=" << param.batchDims;
|
||||
result << "_eType=" << param.expectedTensor.type;
|
||||
if (param.testcaseName != "") {
|
||||
result << "_eShape=" << param.expectedTensor.shape;
|
||||
result << "_=" << param.testcaseName;
|
||||
} else {
|
||||
result << "_eShape=" << param.expectedTensor.shape;
|
||||
}
|
||||
return result.str();
|
||||
}
|
||||
|
||||
private:
|
||||
static std::shared_ptr<Function> CreateFunction(const GatherParamsV7& params) {
|
||||
const auto P = std::make_shared<op::v0::Parameter>(params.dataTensor.type, params.dataTensor.shape);
|
||||
const auto I = std::make_shared<op::v0::Parameter>(params.indicesTensor.type, params.indicesTensor.shape);
|
||||
const auto A = op::v0::Constant::create(params.axisTensor.type, params.axisTensor.shape,
|
||||
params.axisTensor.data.data());
|
||||
const auto G = std::make_shared<op::v7::Gather>(P, I, A, params.batchDims);
|
||||
const auto f = std::make_shared<Function>(G, ParameterVector{P, I});
|
||||
return f;
|
||||
}
|
||||
};
|
||||
|
||||
TEST_P(ReferenceGatherTestV7, CompareWithRefs) {
|
||||
Exec();
|
||||
}
|
||||
|
||||
template <element::Type_t ET, element::Type_t ET_I, element::Type_t ET_A>
|
||||
std::vector<GatherParams> generateParams() {
|
||||
using T = typename element_type_traits<ET>::value_type;
|
||||
using T_I = typename element_type_traits<ET_I>::value_type;
|
||||
using T_A = typename element_type_traits<ET_A>::value_type;
|
||||
std::vector<GatherParams> params {
|
||||
GatherParams(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
10, 11, 20, 21, 30, 31}),
|
||||
Tensor(ET_I, {2, 2, 3, 4}, std::vector<T_I>{
|
||||
0, 1, 1, 2, 0, 1, 1, 2, 0, 1, 1, 2, 0, 1, 1, 2, 0, 1, 1, 2, 0, 1, 1, 2,
|
||||
0, 1, 1, 2, 0, 1, 1, 2, 0, 1, 1, 2, 0, 1, 1, 2, 0, 1, 1, 2, 0, 1, 1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
Tensor(ET, {2, 2, 3, 4, 2}, std::vector<T>{
|
||||
10, 11, 20, 21, 20, 21, 30, 31, 10, 11, 20, 21, 20, 21, 30, 31, 10, 11, 20, 21, 20, 21, 30, 31,
|
||||
10, 11, 20, 21, 20, 21, 30, 31, 10, 11, 20, 21, 20, 21, 30, 31, 10, 11, 20, 21, 20, 21, 30, 31,
|
||||
10, 11, 20, 21, 20, 21, 30, 31, 10, 11, 20, 21, 20, 21, 30, 31, 10, 11, 20, 21, 20, 21, 30, 31,
|
||||
10, 11, 20, 21, 20, 21, 30, 31, 10, 11, 20, 21, 20, 21, 30, 31, 10, 11, 20, 21, 20, 21, 30, 31}),
|
||||
"gather_4d_indices_axis_0"),
|
||||
GatherParams(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
10, 11, 20, 21, 30, 31}),
|
||||
Tensor(ET_I, {2, 2}, std::vector<T_I>{
|
||||
0, 1, 1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
Tensor(ET, {2, 2, 2}, std::vector<T>{
|
||||
10, 11, 20, 21, 20, 21, 30, 31}),
|
||||
"gather_axis_0"),
|
||||
};
|
||||
return params;
|
||||
}
|
||||
|
||||
template <element::Type_t ET, element::Type_t ET_I, element::Type_t ET_A>
|
||||
std::vector<GatherParams> generateParamsFloatValue() {
|
||||
using T = typename element_type_traits<ET>::value_type;
|
||||
using T_I = typename element_type_traits<ET_I>::value_type;
|
||||
using T_A = typename element_type_traits<ET_A>::value_type;
|
||||
std::vector<GatherParams> params {
|
||||
GatherParams(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
Tensor(ET_I, {2, 2, 3, 4}, std::vector<T_I>{
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
|
||||
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
Tensor(ET, {2, 2, 3, 4, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
"gather_4d_indices_axis_0_2d_input"),
|
||||
GatherParams(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
Tensor(ET_I, {2, 3, 4}, std::vector<T_I>{
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
Tensor(ET, {2, 3, 4, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
"gather_3d_indices_axis_0_2d_input"),
|
||||
GatherParams(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
Tensor(ET_I, {2, 2}, std::vector<T_I>{
|
||||
0, 1, 1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
Tensor(ET, {2, 2, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
"gather_2d_indices_axis_0_2d_input"),
|
||||
GatherParams(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
Tensor(ET_I, {2, 2}, std::vector<T_I>{
|
||||
0, -2, 1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
Tensor(ET, {2, 2, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
"gather_2d_negative_and_positive_indices_axis_0_2d_input"),
|
||||
GatherParams(
|
||||
Tensor(ET, {3}, std::vector<T>{
|
||||
1.0f, 2.0f, 3.0f}),
|
||||
Tensor(ET_I, {2}, std::vector<T_I>{
|
||||
1, 0}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
Tensor(ET, {2}, std::vector<T>{
|
||||
2.0f, 1.0f}),
|
||||
"gather_1d_indices_axis_0_1d_input"),
|
||||
GatherParams(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
1.0f, 1.1f, 2.0f, 2.1f, 3.0f, 3.1f}),
|
||||
Tensor(ET_I, {}, std::vector<T_I>{1}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
Tensor(ET, {2}, std::vector<T>{
|
||||
2.0f, 2.1f}),
|
||||
"gather_scalar_indices_axis_0_2d_input"),
|
||||
GatherParams(
|
||||
Tensor(ET, {3, 3}, std::vector<T>{
|
||||
1.0f, 1.1f, 1.2f,
|
||||
2.0f, 2.1f, 2.2f,
|
||||
3.0f, 3.1f, 3.2f}),
|
||||
Tensor(ET_I, {1, 2}, std::vector<T_I>{
|
||||
0, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{1}),
|
||||
Tensor(ET, {3, 1, 2}, std::vector<T>{
|
||||
1.0f, 1.2f,
|
||||
2.0f, 2.2f,
|
||||
3.0f, 3.2f}),
|
||||
"gather_2d_indices_axis_1_2d_input"),
|
||||
GatherParams(
|
||||
Tensor(ET, {2, 2, 3, 3}, std::vector<T>{
|
||||
1.0f, 1.1f, 1.2f,
|
||||
2.0f, 2.1f, 2.2f,
|
||||
3.0f, 3.1f, 3.2f,
|
||||
|
||||
11.0f, 11.1f, 11.2f,
|
||||
12.0f, 12.1f, 12.2f,
|
||||
13.0f, 13.1f, 13.2f,
|
||||
|
||||
|
||||
101.0f, 101.1f, 101.2f,
|
||||
102.0f, 102.1f, 102.2f,
|
||||
103.0f, 103.1f, 103.2f,
|
||||
|
||||
111.0f, 111.1f, 111.2f,
|
||||
112.0f, 112.1f, 112.2f,
|
||||
113.0f, 113.1f, 113.2f}),
|
||||
Tensor(ET_I, {2}, std::vector<T_I>{
|
||||
0, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{2}),
|
||||
Tensor(ET, {2, 2, 2, 3}, std::vector<T>{
|
||||
1.0f, 1.1f, 1.2f,
|
||||
3.0f, 3.1f, 3.2f,
|
||||
|
||||
11.0f, 11.1f, 11.2f,
|
||||
13.0f, 13.1f, 13.2f,
|
||||
|
||||
|
||||
101.0f, 101.1f, 101.2f,
|
||||
103.0f, 103.1f, 103.2f,
|
||||
|
||||
111.0f, 111.1f, 111.2f,
|
||||
113.0f, 113.1f, 113.2f}),
|
||||
"gather_1d_indices_axis_2_4d_input"),
|
||||
GatherParams(
|
||||
Tensor(ET, {3, 3}, std::vector<T>{
|
||||
1.0f, 1.1f, 1.2f, 2.0f, 2.1f, 2.2f, 3.0f, 3.1f, 3.2f}),
|
||||
Tensor(ET_I, {}, std::vector<T_I>{0}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{1}),
|
||||
Tensor(ET, {3}, std::vector<T>{
|
||||
1.0f, 2.0f, 3.0f}),
|
||||
"gather_scalar_indices_axis_1_2d_input"),
|
||||
};
|
||||
return params;
|
||||
}
|
||||
|
||||
std::vector<GatherParams> generateCombinedParams() {
|
||||
const std::vector<std::vector<GatherParams>> generatedParams {
|
||||
generateParams<element::Type_t::i8, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParams<element::Type_t::i16, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParams<element::Type_t::i32, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParams<element::Type_t::i64, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParams<element::Type_t::u8, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParams<element::Type_t::u16, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParams<element::Type_t::u32, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParams<element::Type_t::u64, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsFloatValue<element::Type_t::bf16, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsFloatValue<element::Type_t::f16, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsFloatValue<element::Type_t::f32, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsFloatValue<element::Type_t::f64, element::Type_t::i32, element::Type_t::i64>(),
|
||||
};
|
||||
std::vector<GatherParams> combinedParams;
|
||||
|
||||
for (const auto& params : generatedParams) {
|
||||
combinedParams.insert(combinedParams.end(), params.begin(), params.end());
|
||||
}
|
||||
return combinedParams;
|
||||
}
|
||||
|
||||
INSTANTIATE_TEST_SUITE_P(smoke_Gather_With_Hardcoded_Refs, ReferenceGatherTest,
|
||||
testing::ValuesIn(generateCombinedParams()), ReferenceGatherTest::getTestCaseName);
|
||||
|
||||
template <element::Type_t ET, element::Type_t ET_I, element::Type_t ET_A>
|
||||
std::vector<GatherParamsV7> generateParamsV7() {
|
||||
using T = typename element_type_traits<ET>::value_type;
|
||||
using T_I = typename element_type_traits<ET_I>::value_type;
|
||||
using T_A = typename element_type_traits<ET_A>::value_type;
|
||||
std::vector<GatherParamsV7> params {
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {3}, std::vector<T>{
|
||||
1, 2, 3}),
|
||||
Tensor(ET_I, {2}, std::vector<T_I>{
|
||||
2, 0}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
0,
|
||||
Tensor(ET, {2}, std::vector<T>{
|
||||
3, 1}),
|
||||
"gather_v7_1d"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
10, 11, 20, 21, 30, 31}),
|
||||
Tensor(ET_I, {2, 2}, std::vector<T_I>{
|
||||
0, 1, 1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
0,
|
||||
Tensor(ET, {2, 2, 2}, std::vector<T>{
|
||||
10, 11, 20, 21, 20, 21, 30, 31}),
|
||||
"gather_v7_axis_0"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {2, 3}, std::vector<T>{
|
||||
1, 2, 3,
|
||||
4, 5, 6}),
|
||||
Tensor(ET_I, {2, 2, 2}, std::vector<T_I>{
|
||||
0, 1,
|
||||
1, 2,
|
||||
|
||||
2, 0,
|
||||
1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{1}),
|
||||
1,
|
||||
Tensor(ET, {2, 2, 2}, std::vector<T>{
|
||||
1, 2,
|
||||
2, 3,
|
||||
|
||||
6, 4,
|
||||
5, 6}),
|
||||
"gather_v7_data_int32_3d_indices_axis_1_batch_dims_1"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {2, 5}, std::vector<T>{
|
||||
1, 2, 3, 4, 5,
|
||||
6, 7, 8, 9, 10}),
|
||||
Tensor(ET_I, {2, 3}, std::vector<T_I>{
|
||||
0, 0, 4,
|
||||
4, 0, 0}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{1}),
|
||||
1,
|
||||
Tensor(ET, {2, 3}, std::vector<T>{
|
||||
1, 1, 5,
|
||||
10, 6, 6}),
|
||||
"gather_v7_data_int32_2d_indices_axis_1_batch_dims_1"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {2, 5}, std::vector<T>{
|
||||
1, 2, 3, 4, 5,
|
||||
6, 7, 8, 9, 10}),
|
||||
Tensor(ET_I, {2, 3}, std::vector<T_I>{
|
||||
0, 0, 4,
|
||||
4, 0, 0}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{1}),
|
||||
-1,
|
||||
Tensor(ET, {2, 3}, std::vector<T>{
|
||||
1, 1, 5,
|
||||
10, 6, 6}),
|
||||
"gather_v7_data_int32_2d_indices_axis_1_negative_batch_dims"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {2, 1, 5, 4}, 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, 28,
|
||||
29, 30, 31, 32,
|
||||
33, 34, 35, 36,
|
||||
37, 38, 39, 40}),
|
||||
Tensor(ET_I, {2, 3}, std::vector<T_I>{
|
||||
1, 2, 4,
|
||||
4, 3, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{2}),
|
||||
1,
|
||||
Tensor(ET, {2, 1, 3, 4}, std::vector<T>{
|
||||
5, 6, 7, 8,
|
||||
9, 10, 11, 12,
|
||||
17, 18, 19, 20,
|
||||
|
||||
37, 38, 39, 40,
|
||||
33, 34, 35, 36,
|
||||
29, 30, 31, 32}),
|
||||
"gather_v7_4d_data_axis_2_batch_dims_1_int32"),
|
||||
};
|
||||
return params;
|
||||
}
|
||||
|
||||
template <>
|
||||
std::vector<GatherParamsV7> generateParamsV7<element::boolean, element::i32, element::i64>() {
|
||||
std::vector<GatherParamsV7> params {
|
||||
GatherParamsV7(
|
||||
Tensor(element::boolean, {3, 2}, std::vector<char>{
|
||||
1, 1, 1, 0, 0, 1}),
|
||||
Tensor(element::i32, {2, 2}, std::vector<int32_t>{
|
||||
0, 1, 1, 2}),
|
||||
Tensor(element::i64, {}, std::vector<int64_t>{0}),
|
||||
0,
|
||||
Tensor(element::boolean, {2, 2, 2}, std::vector<char>{
|
||||
1, 1, 1, 0, 1, 0, 0, 1}),
|
||||
"gather_v7_axis_0_bool"),
|
||||
};
|
||||
return params;
|
||||
}
|
||||
|
||||
template <element::Type_t ET, element::Type_t ET_I, element::Type_t ET_A>
|
||||
std::vector<GatherParamsV7> generateParamsFloatValueV7() {
|
||||
using T = typename element_type_traits<ET>::value_type;
|
||||
using T_I = typename element_type_traits<ET_I>::value_type;
|
||||
using T_A = typename element_type_traits<ET_A>::value_type;
|
||||
std::vector<GatherParamsV7> params {
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
Tensor(ET_I, {2, 2, 3, 4}, std::vector<T_I>{
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
|
||||
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
0,
|
||||
Tensor(ET, {2, 2, 3, 4, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
"gather_v7_4d_indices_axis_0_2d_input"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
Tensor(ET_I, {2, 3, 4}, std::vector<T_I>{
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2,
|
||||
0, 1, 1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
0,
|
||||
Tensor(ET, {2, 3, 4, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f,
|
||||
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
"gather_v7_3d_indices_axis_0_2d_input"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
Tensor(ET_I, {2, 2}, std::vector<T_I>{
|
||||
0, 1, 1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
0,
|
||||
Tensor(ET, {2, 2, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
"gather_v7_2d_indices_axis_0_2d_input"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
Tensor(ET_I, {2, 2}, std::vector<T_I>{
|
||||
0, -2, 1, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
0,
|
||||
Tensor(ET, {2, 2, 2}, std::vector<T>{
|
||||
1.0f, 1.1f,
|
||||
2.0f, 2.1f,
|
||||
|
||||
2.0f, 2.1f,
|
||||
3.0f, 3.1f}),
|
||||
"gather_v7_2d_negative_and_positive_indices_axis_0_2d_input"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {3}, std::vector<T>{
|
||||
1.0f, 2.0f, 3.0f}),
|
||||
Tensor(ET_I, {2}, std::vector<T_I>{
|
||||
1, 0}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
0,
|
||||
Tensor(ET, {2}, std::vector<T>{
|
||||
2.0f, 1.0f}),
|
||||
"gather_v7_1d_indices_axis_0_1d_input"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {3, 2}, std::vector<T>{
|
||||
1.0f, 1.1f, 2.0f, 2.1f, 3.0f, 3.1f}),
|
||||
Tensor(ET_I, {}, std::vector<T_I>{1}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{0}),
|
||||
0,
|
||||
Tensor(ET, {2}, std::vector<T>{
|
||||
2.0f, 2.1f}),
|
||||
"gather_v7_scalar_indices_axis_0_2d_input"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {3, 3}, std::vector<T>{
|
||||
1.0f, 1.1f, 1.2f,
|
||||
2.0f, 2.1f, 2.2f,
|
||||
3.0f, 3.1f, 3.2f}),
|
||||
Tensor(ET_I, {1, 2}, std::vector<T_I>{
|
||||
0, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{1}),
|
||||
0,
|
||||
Tensor(ET, {3, 1, 2}, std::vector<T>{
|
||||
1.0f, 1.2f,
|
||||
2.0f, 2.2f,
|
||||
3.0f, 3.2f}),
|
||||
"gather_v7_2d_indices_axis_1_2d_input"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {2, 2, 3, 3}, std::vector<T>{
|
||||
1.0f, 1.1f, 1.2f,
|
||||
2.0f, 2.1f, 2.2f,
|
||||
3.0f, 3.1f, 3.2f,
|
||||
|
||||
11.0f, 11.1f, 11.2f,
|
||||
12.0f, 12.1f, 12.2f,
|
||||
13.0f, 13.1f, 13.2f,
|
||||
|
||||
|
||||
101.0f, 101.1f, 101.2f,
|
||||
102.0f, 102.1f, 102.2f,
|
||||
103.0f, 103.1f, 103.2f,
|
||||
|
||||
111.0f, 111.1f, 111.2f,
|
||||
112.0f, 112.1f, 112.2f,
|
||||
113.0f, 113.1f, 113.2f}),
|
||||
Tensor(ET_I, {2}, std::vector<T_I>{
|
||||
0, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{2}),
|
||||
0,
|
||||
Tensor(ET, {2, 2, 2, 3}, std::vector<T>{
|
||||
1.0f, 1.1f, 1.2f,
|
||||
3.0f, 3.1f, 3.2f,
|
||||
|
||||
11.0f, 11.1f, 11.2f,
|
||||
13.0f, 13.1f, 13.2f,
|
||||
|
||||
|
||||
101.0f, 101.1f, 101.2f,
|
||||
103.0f, 103.1f, 103.2f,
|
||||
|
||||
111.0f, 111.1f, 111.2f,
|
||||
113.0f, 113.1f, 113.2f}),
|
||||
"gather_v7_1d_indices_axis_2_4d_input"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {3, 3}, std::vector<T>{
|
||||
1.0f, 1.1f, 1.2f, 2.0f, 2.1f, 2.2f, 3.0f, 3.1f, 3.2f}),
|
||||
Tensor(ET_I, {}, std::vector<T_I>{0}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{1}),
|
||||
0,
|
||||
Tensor(ET, {3}, std::vector<T>{
|
||||
1.0f, 2.0f, 3.0f}),
|
||||
"gather_v7_scalar_indices_axis_1_2d_input"),
|
||||
GatherParamsV7(
|
||||
Tensor(ET, {2, 5, 2}, std::vector<T>{
|
||||
1.0f, 2.0f,
|
||||
3.0f, 4.0f,
|
||||
5.0f, 6.0f,
|
||||
7.0f, 8.0f,
|
||||
9.0f, 10.0f,
|
||||
|
||||
11.0f, 12.0f,
|
||||
13.0f, 14.0f,
|
||||
15.0f, 16.0f,
|
||||
17.0f, 18.0f,
|
||||
19.0f, 20.0f}),
|
||||
Tensor(ET_I, {2, 2, 3}, std::vector<T_I>{
|
||||
0, 0, 4,
|
||||
4, 0, 0,
|
||||
|
||||
1, 2, 4,
|
||||
4, 3, 2}),
|
||||
Tensor(ET_A, {}, std::vector<T_A>{1}),
|
||||
1,
|
||||
Tensor(ET, {2, 2, 3, 2}, std::vector<T>{
|
||||
1.0f, 2.0f,
|
||||
1.0f, 2.0f,
|
||||
9.0f, 10.0f,
|
||||
|
||||
9.0f, 10.0f,
|
||||
1.0f, 2.0f,
|
||||
1.0f, 2.0f,
|
||||
|
||||
|
||||
13.0f, 14.0f,
|
||||
15.0f, 16.0f,
|
||||
19.0f, 20.0f,
|
||||
|
||||
19.0f, 20.0f,
|
||||
17.0f, 18.0f,
|
||||
15.0f, 16.0f}),
|
||||
"gather_v7_3d_indices_axis_1_batch_dims_1"),
|
||||
};
|
||||
return params;
|
||||
}
|
||||
|
||||
std::vector<GatherParamsV7> generateCombinedParamsV7() {
|
||||
const std::vector<std::vector<GatherParamsV7>> generatedParams {
|
||||
generateParamsV7<element::Type_t::boolean, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsV7<element::Type_t::i8, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsV7<element::Type_t::i16, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsV7<element::Type_t::i32, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsV7<element::Type_t::i64, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsV7<element::Type_t::u8, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsV7<element::Type_t::u16, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsV7<element::Type_t::u32, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsV7<element::Type_t::u64, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsFloatValueV7<element::Type_t::bf16, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsFloatValueV7<element::Type_t::f16, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsFloatValueV7<element::Type_t::f32, element::Type_t::i32, element::Type_t::i64>(),
|
||||
generateParamsFloatValueV7<element::Type_t::f64, element::Type_t::i32, element::Type_t::i64>(),
|
||||
};
|
||||
std::vector<GatherParamsV7> combinedParams;
|
||||
|
||||
for (const auto& params : generatedParams) {
|
||||
combinedParams.insert(combinedParams.end(), params.begin(), params.end());
|
||||
}
|
||||
return combinedParams;
|
||||
}
|
||||
|
||||
INSTANTIATE_TEST_SUITE_P(smoke_Gather_With_Hardcoded_Refs, ReferenceGatherTestV7,
|
||||
testing::ValuesIn(generateCombinedParamsV7()), ReferenceGatherTestV7::getTestCaseName);
|
||||
} // namespace
|
||||
1099
docs/template_plugin/tests/functional/op_reference/pad.cpp
Normal file
1099
docs/template_plugin/tests/functional/op_reference/pad.cpp
Normal file
File diff suppressed because it is too large
Load Diff
@ -61,6 +61,20 @@ std::vector<std::string> disabledTestPatterns() {
|
||||
R"(.*ReferenceGatherElementsTestNegative.*)",
|
||||
// CVS-64052
|
||||
R"(.*ReferenceStridedSliceLayerTest.*strided_slice_stride_optional_dynamic)",
|
||||
// CVS-64017
|
||||
R"(.*ReferenceGatherTest.*dType=i16.*)",
|
||||
R"(.*ReferenceGatherTest.*dType=u16.*)",
|
||||
R"(.*ReferenceGatherTest.*dType=bf16.*)",
|
||||
R"(.*ReferenceGatherTest.*dType=f64.*)",
|
||||
// CVS-64110
|
||||
R"(.*ReferenceGatherTestV7.*dType=i16.*)",
|
||||
R"(.*ReferenceGatherTestV7.*dType=u16.*)",
|
||||
R"(.*ReferenceGatherTestV7.*dType=bf16.*)",
|
||||
R"(.*ReferenceGatherTestV7.*dType=f64.*)",
|
||||
// CVS-64037
|
||||
R"(.*ReferencePadTest.*pad_exterior_2d_0x0)",
|
||||
R"(.*ReferencePadTest.*pad_exterior_2d_0x3)",
|
||||
R"(.*ReferencePadTest.*pad_exterior_2d_3x0)",
|
||||
};
|
||||
|
||||
#ifdef _WIN32
|
||||
|
||||
@ -487,7 +487,6 @@ set(MULTI_TEST_SRC
|
||||
backend/dynamic.in.cpp
|
||||
backend/experimental_detectron_prior_grid.in.cpp
|
||||
backend/function_name.in.cpp
|
||||
backend/gather.in.cpp
|
||||
backend/interpolate.in.cpp
|
||||
backend/matrix_nms.in.cpp
|
||||
backend/multiclass_nms.in.cpp
|
||||
@ -496,7 +495,6 @@ set(MULTI_TEST_SRC
|
||||
backend/node_name.in.cpp
|
||||
backend/non_max_suppression.in.cpp
|
||||
backend/one_hot.in.cpp
|
||||
backend/pad.in.cpp
|
||||
backend/recurrent_cells.in.cpp
|
||||
backend/region_yolo.in.cpp
|
||||
backend/roll.in.cpp
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue
Block a user