aboutsummaryrefslogtreecommitdiffhomepage
path: root/unsupported/test/cxx11_tensor_morphing.cpp
diff options
context:
space:
mode:
authorGravatar Benoit Steiner <benoit.steiner.goog@gmail.com>2016-06-01 21:13:00 -0700
committerGravatar Benoit Steiner <benoit.steiner.goog@gmail.com>2016-06-01 21:13:00 -0700
commitc3cada38e28359e877f6c813d4312beffabefb5a (patch)
tree93252c9bca7549ae54f37f12333ad6c4ffd639d3 /unsupported/test/cxx11_tensor_morphing.cpp
parent360e311b66317fcd8dcb6f97c186252a4de202e2 (diff)
Speedup a test
Diffstat (limited to 'unsupported/test/cxx11_tensor_morphing.cpp')
-rw-r--r--unsupported/test/cxx11_tensor_morphing.cpp21
1 files changed, 9 insertions, 12 deletions
diff --git a/unsupported/test/cxx11_tensor_morphing.cpp b/unsupported/test/cxx11_tensor_morphing.cpp
index 2465ce9f2..c575d3fdc 100644
--- a/unsupported/test/cxx11_tensor_morphing.cpp
+++ b/unsupported/test/cxx11_tensor_morphing.cpp
@@ -324,44 +324,40 @@ static void test_strided_slice()
typedef Tensor<float, 2, DataLayout> Tensor2f;
typedef Eigen::DSizes<Eigen::DenseIndex, 2> Index2;
Tensor<float, 5, DataLayout> tensor(2,3,5,7,11);
+ Tensor<float, 2, DataLayout> tensor2(7,11);
tensor.setRandom();
+ tensor2.setRandom();
- if(true) {
- Tensor<float, 2, DataLayout> tensor(7,11);
- tensor.setRandom();
+ if (true) {
Tensor2f slice(2,3);
Index2 strides(-2,-1);
Index2 indicesStart(5,7);
Index2 indicesStop(0,4);
- slice = tensor.stridedSlice(indicesStart, indicesStop, strides);
+ slice = tensor2.stridedSlice(indicesStart, indicesStop, strides);
for (int j = 0; j < 2; ++j) {
for (int k = 0; k < 3; ++k) {
- VERIFY_IS_EQUAL(slice(j,k), tensor(5-2*j,7-k));
+ VERIFY_IS_EQUAL(slice(j,k), tensor2(5-2*j,7-k));
}
}
}
if(true) {
- Tensor<float, 2, DataLayout> tensor(7,11);
- tensor.setRandom();
Tensor2f slice(0,1);
Index2 strides(1,1);
Index2 indicesStart(5,4);
Index2 indicesStop(5,5);
- slice = tensor.stridedSlice(indicesStart, indicesStop, strides);
+ slice = tensor2.stridedSlice(indicesStart, indicesStop, strides);
}
if(true) { // test clamped degenerate interavls
- Tensor<float, 2, DataLayout> tensor(7,11);
- tensor.setRandom();
Tensor2f slice(7,11);
Index2 strides(1,-1);
Index2 indicesStart(-3,20); // should become 0,10
Index2 indicesStop(20,-11); // should become 11, -1
- slice = tensor.stridedSlice(indicesStart, indicesStop, strides);
+ slice = tensor2.stridedSlice(indicesStart, indicesStop, strides);
for (int j = 0; j < 7; ++j) {
for (int k = 0; k < 11; ++k) {
- VERIFY_IS_EQUAL(slice(j,k), tensor(j,10-k));
+ VERIFY_IS_EQUAL(slice(j,k), tensor2(j,10-k));
}
}
}
@@ -389,6 +385,7 @@ static void test_strided_slice()
}
}
}
+
if(true) {
Tensor5f slice(1,1,2,2,3);
Index5 strides3(1, 1, -2, 1, -1);