Home
last modified time | relevance | path

Searched defs:iendH (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveAveragePooling.cu81 int iendH = END_IND(oh, osizeH, isizeH); in adaptive_average_pool() local
129 int iendH = isizeH; in adaptive_average_gradinput() local
196 int iendH = END_IND(oh, osizeH, isizeH); in atomic_adaptive_average_gradinput() local
281 int iendH = END_IND_INT(oh, osizeH, isizeH); in C10_LAUNCH_BOUNDS_1() local
398 index_t iendH = ::min(istartH+iH, isizeH); in C10_LAUNCH_BOUNDS_1() local
H A DAdaptiveAveragePooling3d.cu88 int iendH = end_index(oh, osizeH, isizeH); in adaptiveaveragepool() local
164 int iendH = isizeH; in adaptiveaveragegradinput() local
286 int iendH = end_index(oh, osizeH, isizeH); in atomicadaptiveaveragegradinput() local
H A DAdaptiveMaxPooling2d.cu74 int iendH = end_index(oh, osizeH, isizeH); in adaptivemaxpool() local
H A DAdaptiveMaxPooling3d.cu86 int iendH = end_index(oh, osizeH, isizeH); in adaptivemaxpool() local
H A DDilatedMaxPool2d.cu244 int iendH = ::min(static_cast<int64_t>(istartH)+iH, height); in C10_LAUNCH_BOUNDS_1() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAdaptiveAveragePooling3d.cpp51 auto iendH = end_index(oh, osizeH, isizeH); in adaptive_avg_pool3d_out_frame() local
202 auto iendH = end_index(oh, osizeH, isizeH); in adaptive_avg_pool3d_backward_out_frame() local
H A DAdaptiveMaxPooling3d.cpp113 int64_t iendH = end_index(oh, osizeH, isizeH); in adaptive_max_pool3d_single_out_frame() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DAdaptiveAveragePooling.cpp86 int iendH = end_index(oh, osizeH, isizeH); in adaptive_avg_pool_single_out_frame() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp1849 int iendH = (int)std::ceil((float)((oh + 1) * isizeH) / osizeH); local