Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveMaxPooling2d.cu142 const T *ptr_gradOutput = gradOutput + oh*osizeW + ow; in adaptivemaxgradinput() local
189 const T *ptr_gradOutput = gradOutput + oh*osizeW + ow; in atomicadaptivemaxgradinput() local
H A DAdaptiveMaxPooling3d.cu193 const T *ptr_gradOutput = gradOutput_dt + oh*osizeW + ow; in adaptivemaxgradinput() local
269 const T *ptr_gradOutput = gradOutput_dt + oh*osizeW + ow; in atomicadaptivemaxgradinput() local
H A DAdaptiveAveragePooling3d.cu195 const scalar_t *ptr_gradOutput = gradOutput_dt; in adaptiveaveragegradinput() local
296 const scalar_t *ptr_gradOutput = gradOutput_nt + oh*osizeW + ow; in atomicadaptiveaveragegradinput() local
H A DAdaptiveAveragePooling.cu207 const T *ptr_gradOutput = gradOutput + oh*osizeW + ow; in atomic_adaptive_average_gradinput() local
413 const scalar_t* ptr_gradOutput = gradOutput + oh*ostrideH + ow*ostrideW; in C10_LAUNCH_BOUNDS_1() local