aboutsummaryrefslogtreecommitdiffhomepage
path: root/unsupported/test/cxx11_tensor_image_patch.cpp
diff options
context:
space:
mode:
authorGravatar Benoit Steiner <benoit.steiner.goog@gmail.com>2016-10-27 21:48:21 -0700
committerGravatar Benoit Steiner <benoit.steiner.goog@gmail.com>2016-10-27 21:48:21 -0700
commitd5f88e2357699193f82f0f44f14c8cfaface60a0 (patch)
tree05c72965757b5bb1347c9f6ef1c579e1772adfe9 /unsupported/test/cxx11_tensor_image_patch.cpp
parent0b4b0f11e889674effe698d792caea7d00ba067b (diff)
Sharded the tensor_image_patch test to help it run on low power devices
Diffstat (limited to 'unsupported/test/cxx11_tensor_image_patch.cpp')
-rw-r--r--unsupported/test/cxx11_tensor_image_patch.cpp24
1 files changed, 12 insertions, 12 deletions
diff --git a/unsupported/test/cxx11_tensor_image_patch.cpp b/unsupported/test/cxx11_tensor_image_patch.cpp
index 988b01481..475c59651 100644
--- a/unsupported/test/cxx11_tensor_image_patch.cpp
+++ b/unsupported/test/cxx11_tensor_image_patch.cpp
@@ -13,7 +13,7 @@
using Eigen::Tensor;
-static void test_simple_patch()
+void test_simple_patch()
{
Tensor<float, 4> tensor(2,3,5,7);
tensor.setRandom();
@@ -180,7 +180,7 @@ static void test_simple_patch()
}
// Verifies VALID padding (no padding) with incrementing values.
-static void test_patch_padding_valid()
+void test_patch_padding_valid()
{
int input_depth = 3;
int input_rows = 3;
@@ -256,7 +256,7 @@ static void test_patch_padding_valid()
}
// Verifies VALID padding (no padding) with the same value.
-static void test_patch_padding_valid_same_value()
+void test_patch_padding_valid_same_value()
{
int input_depth = 1;
int input_rows = 5;
@@ -329,7 +329,7 @@ static void test_patch_padding_valid_same_value()
}
// Verifies SAME padding.
-static void test_patch_padding_same()
+void test_patch_padding_same()
{
int input_depth = 3;
int input_rows = 4;
@@ -405,7 +405,7 @@ static void test_patch_padding_same()
}
}
-static void test_patch_no_extra_dim()
+void test_patch_no_extra_dim()
{
Tensor<float, 3> tensor(2,3,5);
tensor.setRandom();
@@ -553,7 +553,7 @@ static void test_patch_no_extra_dim()
}
}
-static void test_imagenet_patches()
+void test_imagenet_patches()
{
// Test the code on typical configurations used by the 'imagenet' benchmarks at
// https://github.com/soumith/convnet-benchmarks
@@ -748,10 +748,10 @@ static void test_imagenet_patches()
void test_cxx11_tensor_image_patch()
{
- CALL_SUBTEST(test_simple_patch());
- CALL_SUBTEST(test_patch_no_extra_dim());
- CALL_SUBTEST(test_patch_padding_valid());
- CALL_SUBTEST(test_patch_padding_valid_same_value());
- CALL_SUBTEST(test_patch_padding_same());
- CALL_SUBTEST(test_imagenet_patches());
+ CALL_SUBTEST_1(test_simple_patch());
+ CALL_SUBTEST_2(test_patch_no_extra_dim());
+ CALL_SUBTEST_3(test_patch_padding_valid());
+ CALL_SUBTEST_4(test_patch_padding_valid_same_value());
+ CALL_SUBTEST_5(test_patch_padding_same());
+ CALL_SUBTEST_6(test_imagenet_patches());
}