diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/validation/NEON/Reverse.cpp | 2 | ||||
-rw-r--r-- | tests/validation/fixtures/ReverseFixture.h | 10 |
2 files changed, 6 insertions, 6 deletions
diff --git a/tests/validation/NEON/Reverse.cpp b/tests/validation/NEON/Reverse.cpp index eb0c995ad9..7b5337f14b 100644 --- a/tests/validation/NEON/Reverse.cpp +++ b/tests/validation/NEON/Reverse.cpp @@ -44,7 +44,7 @@ namespace validation namespace { using framework::dataset::make; -auto run_small_dataset = combine(datasets::SmallShapes(), datasets::Tiny1DShapes()); +auto run_small_dataset = combine(datasets::Small3DShapes(), datasets::Tiny1DShapes()); auto run_large_dataset = combine(datasets::LargeShapes(), datasets::Tiny1DShapes()); } // namespace diff --git a/tests/validation/fixtures/ReverseFixture.h b/tests/validation/fixtures/ReverseFixture.h index 3bf4c3e327..d53945191e 100644 --- a/tests/validation/fixtures/ReverseFixture.h +++ b/tests/validation/fixtures/ReverseFixture.h @@ -86,9 +86,9 @@ protected: { library->fill_tensor_uniform(tensor, 0); } - std::vector<int> generate_random_axis(bool use_negative = false) + std::vector<int32_t> generate_random_axis(bool use_negative = false) { - std::vector<int> axis_v; + std::vector<int32_t> axis_v; if(use_negative) { axis_v = { -1, -2, -3, -4 }; @@ -97,7 +97,7 @@ protected: { axis_v = { 0, 1, 2, 3 }; } - axis_v = std::vector<int>(axis_v.begin(), axis_v.begin() + _num_dims); + axis_v = std::vector<int32_t>(axis_v.begin(), axis_v.begin() + _num_dims); std::mt19937 g(library->seed()); std::shuffle(axis_v.begin(), axis_v.end(), g); @@ -134,7 +134,7 @@ protected: { auto axis_data = AccessorType(axis); auto axis_v = generate_random_axis(use_negative_axis); - std::copy(axis_v.begin(), axis_v.begin() + _num_dims, static_cast<int32_t *>(axis_data.data())); + std::copy(axis_v.begin(), axis_v.begin() + axis_shape.total_size(), static_cast<int32_t *>(axis_data.data())); } // Compute function @@ -152,7 +152,7 @@ protected: // Fill reference fill(src); auto axis_v = generate_random_axis(use_negative_axis); - std::copy(axis_v.begin(), axis_v.begin() + _num_dims, axis.data()); + std::copy(axis_v.begin(), axis_v.begin() + axis_shape.total_size(), axis.data()); return reference::reverse<T>(src, axis, use_inverted_axis); } |