faeaf045a9
* 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
49 lines
1.8 KiB
C++
49 lines
1.8 KiB
C++
// Copyright (C) 2018-2021 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#include "engines_util/execute_tools.hpp"
|
|
#include "engines_util/test_case.hpp"
|
|
#include "engines_util/test_engines.hpp"
|
|
#include "gtest/gtest.h"
|
|
#include "ngraph/ngraph.hpp"
|
|
#include "ngraph/runtime/tensor.hpp"
|
|
#include "runtime/backend.hpp"
|
|
#include "util/all_close.hpp"
|
|
#include "util/all_close_f.hpp"
|
|
#include "util/ndarray.hpp"
|
|
#include "util/test_control.hpp"
|
|
|
|
using namespace std;
|
|
using namespace ngraph;
|
|
|
|
static string s_manifest = "${MANIFEST}";
|
|
|
|
using TestEngine = test::ENGINE_CLASS_NAME(${BACKEND_NAME});
|
|
|
|
NGRAPH_TEST(${BACKEND_NAME}, interpolate_down_scales_const_linear) {
|
|
Shape input_shape{1, 1, 2, 4};
|
|
Shape output_shape{1, 1, 1, 2};
|
|
op::v0::InterpolateAttrs attrs;
|
|
attrs.axes = AxisSet{0, 1, 2, 3};
|
|
attrs.mode = "linear";
|
|
attrs.align_corners = false;
|
|
const auto input = make_shared<op::Parameter>(element::f32, input_shape);
|
|
const auto output_shape_input = op::v0::Constant::create(element::i64, {4}, {1, 1, 1, 2});
|
|
std::vector<float> intput_data{1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0};
|
|
|
|
auto interpolate = make_shared<op::v0::Interpolate>(input, output_shape_input, attrs);
|
|
auto f = make_shared<Function>(interpolate, ParameterVector{input});
|
|
|
|
auto backend = runtime::Backend::create("IE_CPU");
|
|
auto input_tensor = backend->create_tensor(element::f32, input_shape);
|
|
auto result_tensor = backend->create_tensor(element::f32, output_shape);
|
|
auto handle = backend->compile(f);
|
|
copy_data(input_tensor, intput_data);
|
|
|
|
handle->call_with_validate({result_tensor}, {input_tensor});
|
|
|
|
vector<float> expected_output{1.0f, 2.66666651f};
|
|
EXPECT_TRUE(test::all_close_f(expected_output, read_vector<float>(result_tensor)));
|
|
}
|