Searched refs:outer_strides (Results 1 – 6 of 6) sorted by relevance
/external/pytorch/aten/src/ATen/native/cpu/ |
D | CopyKernel.cpp | 83 const int64_t *outer_strides = &strides[2]; in reduced_float_copy_kernel() local 124 data[0] += outer_strides[0]; in reduced_float_copy_kernel() 125 data[1] += outer_strides[1]; in reduced_float_copy_kernel() 152 const int64_t *outer_strides = &strides[2]; in reduced_float_copy_kernel() local 190 data[0] += outer_strides[0]; in reduced_float_copy_kernel() 191 data[1] += outer_strides[1]; in reduced_float_copy_kernel()
|
D | Loops.h | 262 static void advance(data_t &data, const int64_t *outer_strides) { 264 data[arg] += outer_strides[arg]; 271 const int64_t *outer_strides = &strides[ntensors]; 276 advance(data, outer_strides); 284 advance(data, outer_strides); 289 advance(data, outer_strides);
|
D | IndexKernel.cpp | 467 const int64_t *outer_strides = &strides[3]; in cpu_hflip_vec() local 519 data_arr[arg] += outer_strides[arg]; in cpu_hflip_vec() 538 const int64_t *outer_strides = &strides[3]; in cpu_vflip_memcpy() local 560 data_arr[arg] += outer_strides[arg]; in cpu_vflip_memcpy() 674 const int64_t *outer_strides = &strides[3]; in cpu_hflip_channels_last_vec() local 679 auto c = -outer_strides[0]; in cpu_hflip_channels_last_vec() 680 TORCH_INTERNAL_ASSERT(c == outer_strides[1]); in cpu_hflip_channels_last_vec() 698 data[arg] += outer_strides[arg]; in cpu_hflip_channels_last_vec()
|
D | Reduce.h | 266 int64_t outer_strides[] = { strides[2], strides[3] }; 269 UNARY_OUTER_LOOP(data, outer_strides, size1, [&] { 277 UNARY_OUTER_LOOP(data, outer_strides, size1, [&] {
|
D | SumKernel.cpp | 561 int64_t outer_strides[] = { strides[2], strides[3] }; in cascade_sum() local 562 UNARY_OUTER_LOOP(data, outer_strides, size1, [&] { in cascade_sum()
|
/external/pytorch/aten/src/ATen/ |
D | TensorIterator.h | 414 const int64_t* outer_strides = &strides[ntensor]; in loop_2d_from_1d() 418 data[arg] += outer_strides[arg]; in loop_2d_from_1d()
|