Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveAveragePooling.cu148 T *ptr_gradInput = gradInput + ih*isizeW + iw; in adaptive_average_gradinput() local
206 T *ptr_gradInput = gradInput + istartH*isizeW + istartW; in atomic_adaptive_average_gradinput() local
423 scalar_t *ptr_gradInput = gradInput + (ih * isizeW + iw) * sizeC; in C10_LAUNCH_BOUNDS_1() local
H A DAdaptiveAveragePooling3d.cu194 scalar_t *ptr_gradInput = gradInput_dt + ih*isizeW + iw; in adaptiveaveragegradinput() local
295 scalar_t *ptr_gradInput = gradInput_nt + istartH*isizeW + istartW; in atomicadaptiveaveragegradinput() local