Home
last modified time | relevance | path

Searched defs:ptr_ind (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveMaxPooling2d.cu86 int64_t *ptr_ind = indices + oh*osizeW + ow; in adaptivemaxpool() local
143 const int64_t *ptr_ind = indices + oh*osizeW + ow; in adaptivemaxgradinput() local
190 const int64_t *ptr_ind = indices + oh*osizeW + ow; in atomicadaptivemaxgradinput() local
H A DAdaptiveMaxPooling3d.cu98 int64_t *ptr_ind = indices_dt + oh*osizeW + ow; in adaptivemaxpool() local
194 const int64_t *ptr_ind = indices_dt + oh*osizeW + ow; in adaptivemaxgradinput() local
270 const int64_t *ptr_ind = indices_dt + oh*osizeW + ow; in atomicadaptivemaxgradinput() local