IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
c5d0dc3e5fc9fd35708ec90942c5077cf8da7399
openvino/ngraph/test/backend/roll.in.cpp
Ilya Churaev faeaf045a9 Graph comparator to ngraph util (#7729) 
* Moved FrameworkNode to nGraph

* Moved graph comparator to ngraph test util

* Fixed build

* Try to fix centos

* Fix export target

* Moved engine utils to separate library

* Removed ONNX util from common library

* Fixed build

* Fixed code style
2021-10-01 07:24:28 +03:00

198 lines
9.9 KiB
C++
Raw Blame History

//*****************************************************************************
// Copyright 2017-2021 Intel Corporation
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//*****************************************************************************
#include "engines_util/execute_tools.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/opsets/opset7.hpp"
#include "ngraph/runtime/tensor.hpp"
#include "ngraph/shape.hpp"
#include "runtime/backend.hpp"
#include "util/all_close.hpp"
#include "util/all_close_f.hpp"
#include "util/test_control.hpp"
using namespace std;
using namespace ngraph;
static string s_manifest = "${MANIFEST}";
NGRAPH_TEST(${BACKEND_NAME}, roll_2d_input) {
Shape shape{4, 3};
auto x = make_shared<opset7::Parameter>(element::f32, shape);
auto shift = make_shared<opset7::Constant>(element::i64, Shape{1}, vector<int64_t>{1});
auto axes = make_shared<opset7::Constant>(element::i64, Shape{1}, vector<int64_t>{0});
auto f = make_shared<Function>(make_shared<opset7::Roll>(x, shift, axes), ParameterVector{x});
auto backend = runtime::Backend::create("${BACKEND_NAME}");
auto x_tensor = backend->create_tensor(element::f32, shape);
copy_data(x_tensor,
vector<float>{50.2907,
70.8054,
-68.3403,
62.6444,
4.9748,
-18.5551,
40.5383,
-15.3859,
-4.5881,
-43.3479,
94.1676,
-95.7097});
auto result = backend->create_tensor(element::f32, shape);
auto handle = backend->compile(f);
handle->call_with_validate({result}, {x_tensor});
EXPECT_TRUE(test::all_close_f((vector<float>{-43.3479,
94.1676,
-95.7097,
50.2907,
70.8054,
-68.3403,
62.6444,
4.9748,
-18.5551,
40.5383,
-15.3859,
-4.5881}),
read_vector<float>(result)));
}
NGRAPH_TEST(${BACKEND_NAME}, roll_2d_input_negative_shift) {
Shape shape{4, 3};
auto x = make_shared<opset7::Parameter>(element::f32, shape);
auto shift = make_shared<opset7::Constant>(element::i32, Shape{2}, vector<int32_t>{-1, 2});
auto axes = make_shared<opset7::Constant>(element::i32, Shape{2}, vector<int32_t>{0, 1});
auto f = make_shared<Function>(make_shared<opset7::Roll>(x, shift, axes), ParameterVector{x});
auto backend = runtime::Backend::create("${BACKEND_NAME}");
auto x_tensor = backend->create_tensor(element::f32, shape);
copy_data(x_tensor,
vector<float>{50.2907,
70.8054,
-68.3403,
62.6444,
4.9748,
-18.5551,
40.5383,
-15.3859,
-4.5881,
-43.3479,
94.1676,
-95.7097});
auto result = backend->create_tensor(element::f32, shape);
auto handle = backend->compile(f);
handle->call_with_validate({result}, {x_tensor});
EXPECT_TRUE(test::all_close_f((vector<float>{4.9748,
-18.5551,
62.6444,
-15.3859,
-4.5881,
40.5383,
94.1676,
-95.7097,
-43.3479,
70.8054,
-68.3403,
50.2907}),
read_vector<float>(result)));
}
NGRAPH_TEST(${BACKEND_NAME}, roll_repeated_axes) {
Shape shape{4, 3};
auto x = make_shared<opset7::Parameter>(element::i64, shape);
auto shift = make_shared<opset7::Constant>(element::i64, Shape{3}, vector<int64_t>{1, 2, 1});
auto axes = make_shared<opset7::Constant>(element::i64, Shape{3}, vector<int64_t>{0, 1, 0});
auto f = make_shared<Function>(make_shared<opset7::Roll>(x, shift, axes), ParameterVector{x});
auto backend = runtime::Backend::create("${BACKEND_NAME}");
auto x_tensor = backend->create_tensor(element::i64, shape);
copy_data(x_tensor, vector<int64_t>{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12});
auto result = backend->create_tensor(element::i64, shape);
auto handle = backend->compile(f);
handle->call_with_validate({result}, {x_tensor});
EXPECT_TRUE(
test::all_close((vector<int64_t>{8, 9, 7, 11, 12, 10, 2, 3, 1, 5, 6, 4}), read_vector<int64_t>(result)));
}
NGRAPH_TEST(${BACKEND_NAME}, roll_3d_input) {
Shape shape{4, 2, 3};
auto x = make_shared<opset7::Parameter>(element::f32, shape);
auto shift = make_shared<opset7::Constant>(element::i64, Shape{3}, vector<int64_t>{2, 1, 3});
auto axes = make_shared<opset7::Constant>(element::i64, Shape{3}, vector<int64_t>{0, 1, 2});
auto f = make_shared<Function>(make_shared<opset7::Roll>(x, shift, axes), ParameterVector{x});
auto backend = runtime::Backend::create("${BACKEND_NAME}");
auto x_tensor = backend->create_tensor(element::f32, shape);
copy_data(x_tensor, vector<float>{94.0773, 33.0599, 58.1724, -20.3640, 54.5372, -54.3023, 10.4662, 11.7532,
-11.7692, 56.4223, -95.3774, 8.8978, 1.9305, 13.8025, 12.0827, 81.4669,
19.5321, -8.9553, -75.3226, 20.8033, 20.7660, 62.7361, 14.9372, -33.0825});
auto result = backend->create_tensor(element::f32, shape);
auto handle = backend->compile(f);
handle->call_with_validate({result}, {x_tensor});
EXPECT_TRUE(
test::all_close_f((vector<float>{81.4669, 19.5321, -8.9553, 1.9305, 13.8025, 12.0827, 62.7361, 14.9372,
-33.0825, -75.3226, 20.8033, 20.7660, -20.3640, 54.5372, -54.3023, 94.0773,
33.0599, 58.1724, 56.4223, -95.3774, 8.8978, 10.4662, 11.7532, -11.7692}),
read_vector<float>(result)));
}
NGRAPH_TEST(${BACKEND_NAME}, roll_3d_input_negative_shift) {
Shape shape{4, 2, 3};
auto x = make_shared<opset7::Parameter>(element::f32, shape);
auto shift = make_shared<opset7::Constant>(element::i32, Shape{3}, vector<int32_t>{-5, 1, 3});
auto axes = make_shared<opset7::Constant>(element::i64, Shape{3}, vector<int64_t>{0, 1, 1});
auto f = make_shared<Function>(make_shared<opset7::Roll>(x, shift, axes), ParameterVector{x});
auto backend = runtime::Backend::create("${BACKEND_NAME}");
auto x_tensor = backend->create_tensor(element::f32, shape);
copy_data(x_tensor, vector<float>{94.0773, 33.0599, 58.1724, -20.3640, 54.5372, -54.3023, 10.4662, 11.7532,
-11.7692, 56.4223, -95.3774, 8.8978, 1.9305, 13.8025, 12.0827, 81.4669,
19.5321, -8.9553, -75.3226, 20.8033, 20.7660, 62.7361, 14.9372, -33.0825});
auto result = backend->create_tensor(element::f32, shape);
auto handle = backend->compile(f);
handle->call_with_validate({result}, {x_tensor});
EXPECT_TRUE(
test::all_close_f((vector<float>{10.4662, 11.7532, -11.7692, 56.4223, -95.3774, 8.8978, 1.9305, 13.8025,
12.0827, 81.4669, 19.5321, -8.9553, -75.3226, 20.8033, 20.7660, 62.7361,
14.9372, -33.0825, 94.0773, 33.0599, 58.1724, -20.3640, 54.5372, -54.3023}),
read_vector<float>(result)));
}
NGRAPH_TEST(${BACKEND_NAME}, roll_negative_axes) {
Shape shape{4, 2, 3};
auto x = make_shared<opset7::Parameter>(element::i32, shape);
auto shift = make_shared<opset7::Constant>(element::i64, Shape{3}, vector<int64_t>{2, -1, -7});
auto axes = make_shared<opset7::Constant>(element::i32, Shape{3}, vector<int32_t>{-1, -1, -2});
auto f = make_shared<Function>(make_shared<opset7::Roll>(x, shift, axes), ParameterVector{x});
auto backend = runtime::Backend::create("${BACKEND_NAME}");
auto x_tensor = backend->create_tensor(element::i32, shape);
copy_data(x_tensor,
vector<int32_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});
auto result = backend->create_tensor(element::i32, shape);
auto handle = backend->compile(f);
handle->call_with_validate({result}, {x_tensor});
EXPECT_TRUE(test::all_close(
(vector<int32_t>{6, 4, 5, 3, 1, 2, 12, 10, 11, 9, 7, 8, 18, 16, 17, 15, 13, 14, 24, 22, 23, 21, 19, 20}),
read_vector<int32_t>(result)));
}
Reference in New Issue View Git Blame Copy Permalink