aboutsummaryrefslogtreecommitdiffhomepage
path: root/unsupported/test/cxx11_tensor_generator.cpp
diff options
context:
space:
mode:
authorGravatar Eugene Zhulenev <ezhulenev@google.com>2019-03-05 16:35:21 -0800
committerGravatar Eugene Zhulenev <ezhulenev@google.com>2019-03-05 16:35:21 -0800
commit5d9a6686ed0e526092973ce2823c15825ffd39b6 (patch)
treed2ab5e3eaf6deb0527c3bd56404a782da79a669b /unsupported/test/cxx11_tensor_generator.cpp
parenta407e022e6046917b1ebeacd54b03fcb079a9706 (diff)
Block evaluation for TensorGeneratorOp
Diffstat (limited to 'unsupported/test/cxx11_tensor_generator.cpp')
-rw-r--r--unsupported/test/cxx11_tensor_generator.cpp6
1 files changed, 3 insertions, 3 deletions
diff --git a/unsupported/test/cxx11_tensor_generator.cpp b/unsupported/test/cxx11_tensor_generator.cpp
index ee5e29b77..6dcf676bb 100644
--- a/unsupported/test/cxx11_tensor_generator.cpp
+++ b/unsupported/test/cxx11_tensor_generator.cpp
@@ -42,11 +42,11 @@ struct Generator2D {
template <int DataLayout>
static void test_2D()
{
- Tensor<float, 2> matrix(5, 7);
+ Tensor<float, 2> matrix(512, 512);
Tensor<float, 2> result = matrix.generate(Generator2D());
- for (int i = 0; i < 5; ++i) {
- for (int j = 0; j < 5; ++j) {
+ for (int i = 0; i < 512; ++i) {
+ for (int j = 0; j < 512; ++j) {
VERIFY_IS_EQUAL(result(i, j), 3*i + 11*j);
}
}