Home
last modified time | relevance | path

Searched defs:oendH (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveAveragePooling.cu70 int oendH = osizeH; in adaptive_average_pool() local
140 int oendH = END_IND(ih, isizeH, osizeH); in adaptive_average_gradinput() local
185 int oendH = osizeH; in atomic_adaptive_average_gradinput() local
273 index_t oendH = ::min(ostartH+oH, osizeH); in C10_LAUNCH_BOUNDS_1() local
406 index_t oendH = END_IND_INT(ih, isizeH, osizeH); in C10_LAUNCH_BOUNDS_1() local
H A DAdaptiveMaxPooling2d.cu63 int oendH = osizeH; in adaptivemaxpool() local
129 int oendH = osizeH; in adaptivemaxgradinput() local
176 int oendH = osizeH; in atomicadaptivemaxgradinput() local
H A DAdaptiveMaxPooling3d.cu59 int oendH = osizeH; in adaptivemaxpool() local
172 int oendH = osizeH; in adaptivemaxgradinput() local
248 int oendH = osizeH; in atomicadaptivemaxgradinput() local
H A DAdaptiveAveragePooling3d.cu64 int oendH = osizeH; in adaptiveaveragepool() local
187 int oendH = end_index(ih, isizeH, osizeH); in adaptiveaveragegradinput() local
262 int oendH = osizeH; in atomicadaptiveaveragegradinput() local
H A DDilatedMaxPool2d.cu125 int oendH = ::min(ostartH+oH, pooled_height); in C10_LAUNCH_BOUNDS_1() local