Searched refs:src_slice_start (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_reverse.cpp | 122 array<ptrdiff_t, 4> src_slice_start; in test_expr_reverse() local
123 src_slice_start[0] = 0; in test_expr_reverse()
124 src_slice_start[1] = 0; in test_expr_reverse()
125 src_slice_start[2] = 0; in test_expr_reverse()
126 src_slice_start[3] = 0; in test_expr_reverse()
128 array<ptrdiff_t, 4> dst_slice_start = src_slice_start; in test_expr_reverse()
133 tensor.slice(src_slice_start, src_slice_dim); in test_expr_reverse()
136 tensor.slice(src_slice_start, src_slice_dim).reverse(dim_rev); in test_expr_reverse()
138 src_slice_start[2] += 1; in test_expr_reverse()
|
| H A D | cxx11_tensor_reverse_sycl.cpp | 165 array<IndexType, 4> src_slice_start; in test_expr_reverse() local
166 src_slice_start[0] = 0; in test_expr_reverse()
167 src_slice_start[1] = 0; in test_expr_reverse()
168 src_slice_start[2] = 0; in test_expr_reverse()
169 src_slice_start[3] = 0; in test_expr_reverse()
171 array<IndexType, 4> dst_slice_start = src_slice_start; in test_expr_reverse()
177 .device(sycl_device) = in_gpu.slice(src_slice_start, src_slice_dim); in test_expr_reverse()
180 in_gpu.slice(src_slice_start, src_slice_dim).reverse(dim_rev); in test_expr_reverse()
182 src_slice_start[2] += 1; in test_expr_reverse()
|
| H A D | cxx11_tensor_shuffling.cpp | 87 array<ptrdiff_t, 4> src_slice_start{{0, 0, 0, 0}}; in test_expr_shuffling() local
93 tensor.slice(src_slice_start, src_slice_dim).shuffle(shuffles); in test_expr_shuffling()
94 src_slice_start[2] += 1; in test_expr_shuffling()
|