Home
last modified time | relevance | path

Searched refs:patch_y (Results 1 – 2 of 2) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_volume_patch.cpp43 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 …]
Dcxx11_tensor_volume_patch_sycl.cpp102 const int patch_y = 3; in test_entire_volume_patch_sycl() local
106 array<IndexType, 5> tensorColMajorRange = {{depth, patch_z, patch_y, patch_x, batch}}; in test_entire_volume_patch_sycl()
107 array<IndexType, 5> tensorRowMajorRange = {{batch, patch_x, patch_y, patch_z, depth}}; in test_entire_volume_patch_sycl()
124 …array<IndexType, 6> patchColMajorTensorRange={{depth,patch_z, patch_y, patch_x, patch_z*patch_y*pa… in test_entire_volume_patch_sycl()
129 …atch_col_major.device(sycl_device)=gpu_col_major.extract_volume_patches(patch_z, patch_y, patch_x); in test_entire_volume_patch_sycl()
141 VERIFY_IS_EQUAL(entire_volume_patch_col_major.dimension(2), patch_y); in test_entire_volume_patch_sycl()
143 VERIFY_IS_EQUAL(entire_volume_patch_col_major.dimension(4), patch_z * patch_y * patch_x); in test_entire_volume_patch_sycl()
149 …array<IndexType, 6> patchRowMajorTensorRange={{batch,patch_z*patch_y*patch_x, patch_x, patch_y, pa… in test_entire_volume_patch_sycl()
154 …atch_row_major.device(sycl_device)=gpu_row_major.extract_volume_patches(patch_z, patch_y, patch_x); in test_entire_volume_patch_sycl()
159 VERIFY_IS_EQUAL(entire_volume_patch_row_major.dimension(1), patch_z * patch_y * patch_x); in test_entire_volume_patch_sycl()
[all …]