Searched defs:input_indexr (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleMoreKernel.cpp | 643 auto input_indexr = [=](int64_t n, int64_t h, int64_t w, int64_t offset){ in cpu_upsample_linear_backward_channels_last() local
692 auto input_indexr = [=](int64_t n, int64_t d, int64_t h, int64_t w, int64_t offset) { in cpu_upsample_linear_backward_channels_last() local
|
| H A D | UpSampleKernel.cpp | 604 auto input_indexr = [=](int64_t n, int64_t h, int64_t w) { in cpu_upsample_linear_channels_last() local
652 auto input_indexr = [=](int64_t n, int64_t d, int64_t h, int64_t w) { in cpu_upsample_linear_channels_last() local
1957 auto input_indexr = [=](int64_t c, int64_t h, int64_t w) { in cpu_upsample_genNd_backward_aa() local
|