devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Block evaluation for TensorGeneratorOp

Browse Source
This commit is contained in:
Eugene Zhulenev
2019-03-05 16:35:21 -08:00
parent a407e022e6
commit 5d9a6686ed
3 changed files with 128 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
unsupported/test/cxx11_tensor_generator.cpp
View File

@@ -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);
}
}