Searched refs:patch_y (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_volume_patch.cpp | 43 const int patch_y = 3; in test_entire_volume_patch() local
47 Tensor<float, 5> tensor(depth, patch_z, patch_y, patch_x, batch); in test_entire_volume_patch()
52 entire_volume_patch = tensor.extract_volume_patches(patch_z, patch_y, patch_x); in test_entire_volume_patch()
55 VERIFY_IS_EQUAL(entire_volume_patch.dimension(2), patch_y); in test_entire_volume_patch()
57 VERIFY_IS_EQUAL(entire_volume_patch.dimension(4), patch_z * patch_y * patch_x); in test_entire_volume_patch()
61 …entire_volume_patch_row_major = tensor_row_major.extract_volume_patches(patch_z, patch_y, patch_x); in test_entire_volume_patch()
63 VERIFY_IS_EQUAL(entire_volume_patch_row_major.dimension(1), patch_z * patch_y * patch_x); in test_entire_volume_patch()
65 VERIFY_IS_EQUAL(entire_volume_patch_row_major.dimension(3), patch_y); in test_entire_volume_patch()
70 const int dy = patch_y - 1; in test_entire_volume_patch()
78 for (int py = 0; py < patch_y; py++) { in test_entire_volume_patch()
[all …]
|