Home
last modified time | relevance | path

Searched defs:i_plane (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveMaxPooling2d.cu56 int i_plane = o_plane; in adaptivemaxpool() local
121 int i_plane = o_plane; in adaptivemaxgradinput() local
169 int i_plane = o_plane; in atomicadaptivemaxgradinput() local
H A DAdaptiveAveragePooling.cu64 int i_plane = o_plane; in adaptive_average_pool() local
122 int i_plane = blockIdx.x; in adaptive_average_gradinput() local
179 int i_plane = o_plane; in atomic_adaptive_average_gradinput() local
H A DAdaptiveAveragePooling3d.cu171 int64_t i_plane = blockIdx.x + offsetZ; in adaptiveaveragegradinput() local