Searched defs:loop2d (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleMoreKernel.cpp | 146 auto loop2d = [&](int64_t begin, int64_t end) { in cpu_upsample_nearest_backward() local
256 auto loop2d = [&](int64_t begin, int64_t end) { in cpu_upsample_nearest_backward_channels_last() local
484 auto loop2d = [&](int64_t begin, int64_t end) { in cpu_upsample_linear_backward() local
627 auto loop2d = [&](int64_t begin, int64_t end) { in cpu_upsample_linear_backward_channels_last() local
|
| H A D | IndexKernel.cpp | 461 auto loop2d = [&](char** base, const int64_t *strides, int64_t size0, int64_t size1) { in cpu_hflip_vec() local
532 auto loop2d = [&](char** base, const int64_t *strides, int64_t size0, int64_t size1) { in cpu_vflip_memcpy() local
668 auto loop2d = [&](char** base, const int64_t *strides, int64_t size0, int64_t size1) { in cpu_hflip_channels_last_vec() local
|
| H A D | UpSampleKernel.cpp | 502 auto loop2d = [&](int64_t begin, int64_t end) { in cpu_upsample_nearest_channels_last() local
598 auto loop2d = [&](int64_t begin, int64_t end) { in cpu_upsample_linear_channels_last() local
1951 auto loop2d = [&](int64_t begin, int64_t end) { in cpu_upsample_genNd_backward_aa() local
|