diff options
author | Eugene Zhulenev <ezhulenev@google.com> | 2019-03-05 16:35:21 -0800 |
---|---|---|
committer | Eugene Zhulenev <ezhulenev@google.com> | 2019-03-05 16:35:21 -0800 |
commit | 5d9a6686ed0e526092973ce2823c15825ffd39b6 (patch) | |
tree | d2ab5e3eaf6deb0527c3bd56404a782da79a669b /unsupported/test/cxx11_tensor_generator.cpp | |
parent | a407e022e6046917b1ebeacd54b03fcb079a9706 (diff) |
Block evaluation for TensorGeneratorOp
Diffstat (limited to 'unsupported/test/cxx11_tensor_generator.cpp')
-rw-r--r-- | unsupported/test/cxx11_tensor_generator.cpp | 6 |
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); } } |