fadd16ce89
* Align ReorgYolo to the spec (vector strides -> int stride) * ReorgYolo ref impl * ReorgYolo evaluate method * ReorgYolo tests * Tests update * Style apply * Add some coments * Code refactor * Comment update * Style apply * Build fix, mark evaluate as override * Revert "Align ReorgYolo to the spec (vector strides -> int stride)" * Use int_executable instead of evaluate * Use char* instead of templates * Code refactor * Comment update * Code review comment * Add constructor aligned with spec * Update shape validation * Update attributes tests * Add type_prop tests * Update backend tests * Add single layer tests * Update the spec * Remove wrong transformation test
102 lines
4.2 KiB
C++
102 lines
4.2 KiB
C++
//*****************************************************************************
|
|
// Copyright 2017-2020 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 <algorithm>
|
|
#include <cinttypes>
|
|
#include <cmath>
|
|
#include <cstdlib>
|
|
#include <random>
|
|
#include <string>
|
|
|
|
// clang-format off
|
|
#ifdef ${BACKEND_NAME}_FLOAT_TOLERANCE_BITS
|
|
#define DEFAULT_FLOAT_TOLERANCE_BITS ${BACKEND_NAME}_FLOAT_TOLERANCE_BITS
|
|
#endif
|
|
|
|
#ifdef ${BACKEND_NAME}_DOUBLE_TOLERANCE_BITS
|
|
#define DEFAULT_DOUBLE_TOLERANCE_BITS ${BACKEND_NAME}_DOUBLE_TOLERANCE_BITS
|
|
#endif
|
|
// clang-format on
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "ngraph/ngraph.hpp"
|
|
#include "util/engine/test_engines.hpp"
|
|
#include "util/test_case.hpp"
|
|
#include "util/test_control.hpp"
|
|
#include "util/type_prop.hpp"
|
|
|
|
using namespace std;
|
|
using namespace ngraph;
|
|
|
|
static string s_manifest = "${MANIFEST}";
|
|
using TestEngine = test::ENGINE_CLASS_NAME(${BACKEND_NAME});
|
|
|
|
NGRAPH_TEST(${BACKEND_NAME}, reorg_yolo_stride_2)
|
|
{
|
|
// in_shape [N,C,H,W]
|
|
const auto in_shape = Shape{1, 8, 4, 4};
|
|
auto p = make_shared<op::Parameter>(element::f32, in_shape);
|
|
size_t stride = 2;
|
|
auto reorg_yolo = make_shared<op::v0::ReorgYolo>(p, Strides{stride});
|
|
auto fun = make_shared<Function>(OutputVector{reorg_yolo}, ParameterVector{p});
|
|
|
|
std::vector<float> inputs(128);
|
|
std::iota(inputs.begin(), inputs.end(), 0);
|
|
std::vector<float> expected_result{
|
|
0, 2, 4, 6, 16, 18, 20, 22, 32, 34, 36, 38, 48, 50, 52, 54,
|
|
64, 66, 68, 70, 80, 82, 84, 86, 96, 98, 100, 102, 112, 114, 116, 118,
|
|
1, 3, 5, 7, 17, 19, 21, 23, 33, 35, 37, 39, 49, 51, 53, 55,
|
|
65, 67, 69, 71, 81, 83, 85, 87, 97, 99, 101, 103, 113, 115, 117, 119,
|
|
8, 10, 12, 14, 24, 26, 28, 30, 40, 42, 44, 46, 56, 58, 60, 62,
|
|
72, 74, 76, 78, 88, 90, 92, 94, 104, 106, 108, 110, 120, 122, 124, 126,
|
|
9, 11, 13, 15, 25, 27, 29, 31, 41, 43, 45, 47, 57, 59, 61, 63,
|
|
73, 75, 77, 79, 89, 91, 93, 95, 105, 107, 109, 111, 121, 123, 125, 127};
|
|
// in_shape [N,C,H,W] -> out_shape [N, C*stride*stride, H/stride, W/stride]
|
|
Shape expected_shape = Shape{
|
|
in_shape[0], in_shape[1] * stride * stride, in_shape[2] / stride, in_shape[3] / stride};
|
|
|
|
auto test_case = test::TestCase<TestEngine>(fun);
|
|
test_case.add_input<float>(inputs);
|
|
test_case.add_expected_output<float>(expected_shape, expected_result);
|
|
test_case.run();
|
|
}
|
|
|
|
NGRAPH_TEST(${BACKEND_NAME}, reorg_yolo_stride_3)
|
|
{
|
|
// in_shape [N,C,H,W]
|
|
const auto in_shape = Shape{1, 9, 3, 3};
|
|
auto p = make_shared<op::Parameter>(element::f32, in_shape);
|
|
size_t stride = 3;
|
|
auto reorg_yolo = make_shared<op::v0::ReorgYolo>(p, Strides{stride});
|
|
auto fun = make_shared<Function>(OutputVector{reorg_yolo}, ParameterVector{p});
|
|
|
|
std::vector<float> inputs(81);
|
|
std::iota(inputs.begin(), inputs.end(), 0);
|
|
std::vector<float> expected_result{
|
|
0, 3, 6, 27, 30, 33, 54, 57, 60, 1, 4, 7, 28, 31, 34, 55, 58, 61, 2, 5, 8,
|
|
29, 32, 35, 56, 59, 62, 9, 12, 15, 36, 39, 42, 63, 66, 69, 10, 13, 16, 37, 40, 43,
|
|
64, 67, 70, 11, 14, 17, 38, 41, 44, 65, 68, 71, 18, 21, 24, 45, 48, 51, 72, 75, 78,
|
|
19, 22, 25, 46, 49, 52, 73, 76, 79, 20, 23, 26, 47, 50, 53, 74, 77, 80};
|
|
// in_shape [N,C,H,W] -> out_shape [N, C*stride*stride, H/stride, W/stride]
|
|
Shape expected_shape = Shape{
|
|
in_shape[0], in_shape[1] * stride * stride, in_shape[2] / stride, in_shape[3] / stride};
|
|
|
|
auto test_case = test::TestCase<TestEngine>(fun);
|
|
test_case.add_input<float>(inputs);
|
|
test_case.add_expected_output<float>(expected_shape, expected_result);
|
|
test_case.run();
|
|
}
|