Searched refs:patch_z (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_volume_patch.cpp | 42 const int patch_z = 2; 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()
54 VERIFY_IS_EQUAL(entire_volume_patch.dimension(1), patch_z); 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()
66 VERIFY_IS_EQUAL(entire_volume_patch_row_major.dimension(4), patch_z); in test_entire_volume_patch()
69 const int dz = patch_z - 1; in test_entire_volume_patch()
77 for (int pz = 0; pz < patch_z; pz++) { in test_entire_volume_patch()
[all …]
|