Home
last modified time | relevance | path

Searched refs:outer_strides (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/aten/src/ATen/native/cpu/
DCopyKernel.cpp83 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()
DLoops.h262 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);
DIndexKernel.cpp467 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()
DReduce.h266 int64_t outer_strides[] = { strides[2], strides[3] };
269 UNARY_OUTER_LOOP(data, outer_strides, size1, [&] {
277 UNARY_OUTER_LOOP(data, outer_strides, size1, [&] {
DSumKernel.cpp561 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/
DTensorIterator.h414 const int64_t* outer_strides = &strides[ntensor]; in loop_2d_from_1d()
418 data[arg] += outer_strides[arg]; in loop_2d_from_1d()