Home
last modified time | relevance | path

Searched defs:ostepH (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveMaxPooling2d.cu64 const int ostepH = blockDim.y*gridDim.y; in adaptivemaxpool() local
130 int ostepH = blockDim.y*gridDim.y; in adaptivemaxgradinput() local
177 int ostepH = blockDim.y*gridDim.y; in atomicadaptivemaxgradinput() local
H A DAdaptiveMaxPooling3d.cu60 int ostepH = gridDim.y * blockDim.y; in adaptivemaxpool() local
173 int ostepH = gridDim.y * blockDim.y; in adaptivemaxgradinput() local
249 int ostepH = gridDim.y * blockDim.y; in atomicadaptivemaxgradinput() local
H A DAdaptiveAveragePooling3d.cu65 int ostepH = gridDim.y * blockDim.y; in adaptiveaveragepool() local
263 int ostepH = gridDim.y * blockDim.y; in atomicadaptiveaveragegradinput() local
H A DAdaptiveAveragePooling.cu71 const int ostepH = blockDim.y*gridDim.y; in adaptive_average_pool() local
186 int ostepH = blockDim.y*gridDim.y; in atomic_adaptive_average_gradinput() local