Home
last modified time | relevance | path

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

/external/eigen/unsupported/test/
Dcxx11_tensor_image_patch_sycl.cpp69 array<IndexType, 5> patchRowMajorTensorRange={{sizeDim4, sizeDim2*sizeDim3, 1, 1, sizeDim1}}; in test_simple_image_patch_sycl() local
70 Tensor<DataType, 5, RowMajor,IndexType> single_patch_row_major(patchRowMajorTensorRange); in test_simple_image_patch_sycl()
73 …,IndexType>> gpu_single_patch_row_major(gpu_data_single_patch_row_major, patchRowMajorTensorRange); in test_simple_image_patch_sycl()
121 patchRowMajorTensorRange={{sizeDim4, sizeDim2*sizeDim3, sizeDim3, sizeDim2, sizeDim1}}; in test_simple_image_patch_sycl()
122 Tensor<DataType, 5, RowMajor,IndexType> entire_image_patch_row_major(patchRowMajorTensorRange); in test_simple_image_patch_sycl()
125 … gpu_entire_image_patch_row_major(gpu_data_entire_image_patch_row_major, patchRowMajorTensorRange); in test_simple_image_patch_sycl()
187 patchRowMajorTensorRange={{sizeDim4, sizeDim2*sizeDim3, 2, 2, sizeDim1}}; in test_simple_image_patch_sycl()
188 Tensor<DataType, 5, RowMajor,IndexType> twod_patch_row_major(patchRowMajorTensorRange); in test_simple_image_patch_sycl()
191 …ajor,IndexType>> gpu_twod_patch_row_major(gpu_data_twod_patch_row_major, patchRowMajorTensorRange); in test_simple_image_patch_sycl()
306 array<IndexType, 5> patchRowMajorTensorRange={{input_batches, 1, ksize, ksize, input_depth }}; in test_patch_padding_valid_sycl() local
[all …]
Dcxx11_tensor_volume_patch_sycl.cpp68 …array<IndexType, 6> patchRowMajorTensorRange={{sizeDim4, sizeDim1*sizeDim2*sizeDim3, 1, 1, 1, size… in test_single_voxel_patch_sycl() local
69 Tensor<DataType, 6, RowMajor,IndexType> single_voxel_patch_row_major(patchRowMajorTensorRange); in test_single_voxel_patch_sycl()
72 … gpu_single_voxel_patch_row_major(gpu_data_single_voxel_patch_row_major, patchRowMajorTensorRange); in test_single_voxel_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() local
150 Tensor<DataType, 6, RowMajor,IndexType> entire_volume_patch_row_major(patchRowMajorTensorRange); in test_entire_volume_patch_sycl()
153 …pu_entire_volume_patch_row_major(gpu_data_entire_volume_patch_row_major, patchRowMajorTensorRange); in test_entire_volume_patch_sycl()