Home
last modified time | relevance | path

Searched refs:tensor_row_major (Results 1 – 3 of 3) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_image_patch.cpp20 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout(); in test_simple_patch() local
21 VERIFY_IS_EQUAL(tensor.dimension(0), tensor_row_major.dimension(3)); in test_simple_patch()
22 VERIFY_IS_EQUAL(tensor.dimension(1), tensor_row_major.dimension(2)); in test_simple_patch()
23 VERIFY_IS_EQUAL(tensor.dimension(2), tensor_row_major.dimension(1)); in test_simple_patch()
24 VERIFY_IS_EQUAL(tensor.dimension(3), tensor_row_major.dimension(0)); in test_simple_patch()
37 single_pixel_patch_row_major = tensor_row_major.extract_image_patches(1, 1); in test_simple_patch()
53 if (tensor_row_major.data()[i] != single_pixel_patch_row_major.data()[i]) { in test_simple_patch()
59 tensor_row_major.data()[i]); in test_simple_patch()
60 VERIFY_IS_EQUAL(tensor.data()[i], tensor_row_major.data()[i]); in test_simple_patch()
76 entire_image_patch_row_major = tensor_row_major.extract_image_patches(3, 5); in test_simple_patch()
[all …]
Dcxx11_tensor_volume_patch.cpp11 Tensor<float, 5, RowMajor> tensor_row_major = tensor.swap_layout(); in test_single_voxel_patch() local
23 single_voxel_patch_row_major = tensor_row_major.extract_volume_patches(1, 1, 1); in test_single_voxel_patch()
33 VERIFY_IS_EQUAL(tensor_row_major.data()[i], single_voxel_patch_row_major.data()[i]); in test_single_voxel_patch()
34 VERIFY_IS_EQUAL(tensor.data()[i], tensor_row_major.data()[i]); in test_single_voxel_patch()
49 Tensor<float, 5, RowMajor> tensor_row_major = tensor.swap_layout(); in test_entire_volume_patch() local
61 …entire_volume_patch_row_major = tensor_row_major.extract_volume_patches(patch_z, patch_y, patch_x); in test_entire_volume_patch()
94 expected_row_major = tensor_row_major(b, eff_x, eff_y, eff_z, d); in test_entire_volume_patch()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DREADME.md1680 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout();
1695 twod_patch_row_major = tensor_row_major.extract_image_patches<2, 2>();