publish master branch snapshot, revision 9df5eb1f84e13a35720a918f88324561222ab114
This commit is contained in:
Alexey Suhov
337
ngraph/test/coordinate.cpp
Normal file
337
ngraph/test/coordinate.cpp
Normal file
@@ -0,0 +1,337 @@
|
||||
//*****************************************************************************
|
||||
// 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 <memory>
|
||||
#include <string>
|
||||
|
||||
#include "gtest/gtest.h"
|
||||
|
||||
#include "ngraph/ngraph.hpp"
|
||||
#include "util/ndarray.hpp"
|
||||
#include "util/test_tools.hpp"
|
||||
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
TEST(coordinate, shape0d)
|
||||
{
|
||||
auto ct = CoordinateTransform({});
|
||||
ASSERT_EQ(shape_size(ct.get_target_shape()), 1);
|
||||
auto it = ct.begin();
|
||||
EXPECT_EQ(*it++, Coordinate({}));
|
||||
EXPECT_TRUE(it == ct.end());
|
||||
}
|
||||
|
||||
TEST(coordinate, shape1d)
|
||||
{
|
||||
auto ct = CoordinateTransform({3});
|
||||
ASSERT_EQ(shape_size(ct.get_target_shape()), 3);
|
||||
auto it = ct.begin();
|
||||
EXPECT_EQ(*it++, Coordinate({0}));
|
||||
EXPECT_EQ(*it++, Coordinate({1}));
|
||||
EXPECT_EQ(*it++, Coordinate({2}));
|
||||
EXPECT_TRUE(it == ct.end());
|
||||
}
|
||||
|
||||
TEST(coordinate, shape2d)
|
||||
{
|
||||
auto ct = CoordinateTransform({2, 3});
|
||||
ASSERT_EQ(shape_size(ct.get_target_shape()), 6);
|
||||
auto it = ct.begin();
|
||||
EXPECT_EQ(*it++, Coordinate({0, 0}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 1}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 2}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 0}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 1}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 2}));
|
||||
EXPECT_TRUE(it == ct.end());
|
||||
}
|
||||
|
||||
TEST(coordinate, shape3d)
|
||||
{
|
||||
auto ct = CoordinateTransform({2, 3, 4});
|
||||
ASSERT_EQ(shape_size(ct.get_target_shape()), 24);
|
||||
auto it = ct.begin();
|
||||
EXPECT_EQ(*it++, Coordinate({0, 0, 0}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 0, 1}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 0, 2}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 0, 3}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 1, 0}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 1, 1}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 1, 2}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 1, 3}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 2, 0}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 2, 1}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 2, 2}));
|
||||
EXPECT_EQ(*it++, Coordinate({0, 2, 3}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 0, 0}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 0, 1}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 0, 2}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 0, 3}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 1, 0}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 1, 1}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 1, 2}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 1, 3}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 2, 0}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 2, 1}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 2, 2}));
|
||||
EXPECT_EQ(*it++, Coordinate({1, 2, 3}));
|
||||
EXPECT_TRUE(it == ct.end());
|
||||
}
|
||||
|
||||
TEST(coordinate, zero_sized_axis)
|
||||
{
|
||||
auto ct = CoordinateTransform({2, 0, 4});
|
||||
ASSERT_EQ(shape_size(ct.get_target_shape()), 0);
|
||||
auto it = ct.begin();
|
||||
EXPECT_TRUE(it == ct.end());
|
||||
}
|
||||
|
||||
TEST(DISABLED_coordinate, random)
|
||||
{
|
||||
auto ct = CoordinateTransform({2, 3, 4});
|
||||
ASSERT_EQ(shape_size(ct.get_target_shape()), 24);
|
||||
auto it = ct.begin();
|
||||
it += 5;
|
||||
EXPECT_EQ(*it, Coordinate({0, 1, 1}));
|
||||
it += -2;
|
||||
EXPECT_EQ(*it, Coordinate({0, 1, 1}));
|
||||
}
|
||||
|
||||
TEST(coordinate, corner)
|
||||
{
|
||||
Shape source_shape{10, 10};
|
||||
Coordinate source_start_corner = Coordinate{3, 3};
|
||||
Coordinate source_end_corner{6, 6};
|
||||
Strides source_strides = Strides(source_shape.size(), 1);
|
||||
AxisVector source_axis_order(source_shape.size());
|
||||
iota(source_axis_order.begin(), source_axis_order.end(), 0);
|
||||
CoordinateDiff target_padding_below = CoordinateDiff(source_shape.size(), 0);
|
||||
CoordinateDiff target_padding_above = CoordinateDiff(source_shape.size(), 0);
|
||||
Strides source_dilation_strides = Strides(source_shape.size(), 1);
|
||||
|
||||
auto ct = CoordinateTransform(source_shape,
|
||||
source_start_corner,
|
||||
source_end_corner,
|
||||
source_strides,
|
||||
source_axis_order,
|
||||
target_padding_below,
|
||||
target_padding_above,
|
||||
source_dilation_strides);
|
||||
|
||||
ASSERT_EQ(shape_size(ct.get_target_shape()), 9);
|
||||
auto it = ct.begin();
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({3, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({3, 4}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({3, 5}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({4, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({4, 4}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({4, 5}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({5, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({5, 4}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({5, 5}));
|
||||
EXPECT_TRUE(it == ct.end());
|
||||
}
|
||||
|
||||
TEST(coordinate, strides)
|
||||
{
|
||||
Shape source_shape{10, 10};
|
||||
Coordinate source_start_corner = Coordinate{0, 0};
|
||||
Coordinate source_end_corner{source_shape};
|
||||
Strides source_strides = Strides({2, 3});
|
||||
AxisVector source_axis_order(source_shape.size());
|
||||
iota(source_axis_order.begin(), source_axis_order.end(), 0);
|
||||
CoordinateDiff target_padding_below = CoordinateDiff(source_shape.size(), 0);
|
||||
CoordinateDiff target_padding_above = CoordinateDiff(source_shape.size(), 0);
|
||||
Strides source_dilation_strides = Strides(source_shape.size(), 1);
|
||||
|
||||
auto ct = CoordinateTransform(source_shape,
|
||||
source_start_corner,
|
||||
source_end_corner,
|
||||
source_strides,
|
||||
source_axis_order,
|
||||
target_padding_below,
|
||||
target_padding_above,
|
||||
source_dilation_strides);
|
||||
|
||||
ASSERT_EQ(shape_size(ct.get_target_shape()), 20);
|
||||
auto it = ct.begin();
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 0}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 6}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 9}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 0}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 6}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 9}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({4, 0}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({4, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({4, 6}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({4, 9}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({6, 0}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({6, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({6, 6}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({6, 9}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({8, 0}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({8, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({8, 6}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({8, 9}));
|
||||
EXPECT_TRUE(it == ct.end());
|
||||
}
|
||||
|
||||
TEST(coordinate, axis_order)
|
||||
{
|
||||
Shape source_shape{3, 2, 4};
|
||||
Coordinate source_start_corner = Coordinate{0, 0, 0};
|
||||
Coordinate source_end_corner{source_shape};
|
||||
Strides source_strides = Strides(source_shape.size(), 1);
|
||||
AxisVector source_axis_order({1, 2, 0});
|
||||
CoordinateDiff target_padding_below = CoordinateDiff(source_shape.size(), 0);
|
||||
CoordinateDiff target_padding_above = CoordinateDiff(source_shape.size(), 0);
|
||||
Strides source_dilation_strides = Strides(source_shape.size(), 1);
|
||||
|
||||
auto ct = CoordinateTransform(source_shape,
|
||||
source_start_corner,
|
||||
source_end_corner,
|
||||
source_strides,
|
||||
source_axis_order,
|
||||
target_padding_below,
|
||||
target_padding_above,
|
||||
source_dilation_strides);
|
||||
|
||||
ASSERT_EQ(shape_size(ct.get_target_shape()), 24);
|
||||
auto it = ct.begin();
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 0, 0}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({1, 0, 0}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 0, 0}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 0, 1}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({1, 0, 1}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 0, 1}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 0, 2}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({1, 0, 2}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 0, 2}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 0, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({1, 0, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 0, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 1, 0}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({1, 1, 0}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 1, 0}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 1, 1}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({1, 1, 1}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 1, 1}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 1, 2}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({1, 1, 2}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 1, 2}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({0, 1, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({1, 1, 3}));
|
||||
EXPECT_EQ(ct.to_source_coordinate(*it++), Coordinate({2, 1, 3}));
|
||||
EXPECT_TRUE(it == ct.end());
|
||||
}
|
||||
|
||||
TEST(DISABLED_coordinate, padding)
|
||||
{
|
||||
Shape source_shape{10, 10};
|
||||
Coordinate source_start_corner = Coordinate{0, 0};
|
||||
Coordinate source_end_corner{source_shape};
|
||||
Strides source_strides = Strides(source_shape.size(), 1);
|
||||
AxisVector source_axis_order(source_shape.size());
|
||||
iota(source_axis_order.begin(), source_axis_order.end(), 0);
|
||||
CoordinateDiff target_padding_below = CoordinateDiff(source_shape.size(), 0);
|
||||
CoordinateDiff target_padding_above = CoordinateDiff(source_shape.size(), 0);
|
||||
Strides source_dilation_strides = Strides(source_shape.size(), 1);
|
||||
|
||||
auto ct = CoordinateTransform(source_shape,
|
||||
source_start_corner,
|
||||
source_end_corner,
|
||||
source_strides,
|
||||
source_axis_order,
|
||||
target_padding_below,
|
||||
target_padding_above,
|
||||
source_dilation_strides);
|
||||
|
||||
// for (const Coordinate& c : ct)
|
||||
// {
|
||||
// cout << c << ", " << ct.to_source_coordinate(c) << endl;
|
||||
// }
|
||||
|
||||
ASSERT_EQ(shape_size(ct.get_target_shape()), 24);
|
||||
auto it = ct.begin();
|
||||
|
||||
EXPECT_TRUE(it == ct.end());
|
||||
}
|
||||
|
||||
TEST(DISABLED_coordinate, dilation)
|
||||
{
|
||||
Shape source_shape{10, 10};
|
||||
Coordinate source_start_corner = Coordinate{0, 0};
|
||||
Coordinate source_end_corner{source_shape};
|
||||
Strides source_strides = Strides(source_shape.size(), 1);
|
||||
AxisVector source_axis_order(source_shape.size());
|
||||
iota(source_axis_order.begin(), source_axis_order.end(), 0);
|
||||
CoordinateDiff target_padding_below = CoordinateDiff(source_shape.size(), 0);
|
||||
CoordinateDiff target_padding_above = CoordinateDiff(source_shape.size(), 0);
|
||||
Strides source_dilation_strides = Strides(source_shape.size(), 1);
|
||||
|
||||
auto ct = CoordinateTransform(source_shape,
|
||||
source_start_corner,
|
||||
source_end_corner,
|
||||
source_strides,
|
||||
source_axis_order,
|
||||
target_padding_below,
|
||||
target_padding_above,
|
||||
source_dilation_strides);
|
||||
|
||||
// for (const Coordinate& c : ct)
|
||||
// {
|
||||
// cout << ct.to_source_coordinate(c) << endl;
|
||||
// }
|
||||
|
||||
ASSERT_EQ(shape_size(ct.get_target_shape()), 24);
|
||||
auto it = ct.begin();
|
||||
|
||||
EXPECT_TRUE(it == ct.end());
|
||||
}
|
||||
|
||||
TEST(benchmark, coordinate)
|
||||
{
|
||||
Shape source_shape{128, 3, 2000, 1000};
|
||||
Coordinate source_start_corner = Coordinate{0, 0, 0, 0};
|
||||
Coordinate source_end_corner{source_shape};
|
||||
Strides source_strides = Strides(source_shape.size(), 1);
|
||||
AxisVector source_axis_order(source_shape.size());
|
||||
iota(source_axis_order.begin(), source_axis_order.end(), 0);
|
||||
CoordinateDiff target_padding_below = CoordinateDiff(source_shape.size(), 0);
|
||||
CoordinateDiff target_padding_above = CoordinateDiff(source_shape.size(), 0);
|
||||
Strides source_dilation_strides = Strides(source_shape.size(), 1);
|
||||
|
||||
stopwatch timer;
|
||||
timer.start();
|
||||
auto ct = CoordinateTransform(source_shape,
|
||||
source_start_corner,
|
||||
source_end_corner,
|
||||
source_strides,
|
||||
source_axis_order,
|
||||
target_padding_below,
|
||||
target_padding_above,
|
||||
source_dilation_strides);
|
||||
|
||||
for (const Coordinate& c : ct)
|
||||
{
|
||||
(void)c;
|
||||
}
|
||||
timer.stop();
|
||||
cout << "time: " << timer.get_milliseconds() << endl;
|
||||
}
|
||||
Reference in New Issue
Block a user