Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveMaxPooling2d.cu241 int blocksH = (int)(16L / sizeD); in TORCH_IMPL_FUNC() local
290 int blocksH = (int)(16L / sizeD); in TORCH_IMPL_FUNC() local
374 int blocksH = (int)(16L / sizeD); in TORCH_IMPL_FUNC() local
435 int blocksH = (int)(16L / sizeD); in TORCH_IMPL_FUNC() local
H A DAdaptiveMaxPooling3d.cu136 int blocksH = std::max((int)(16L / totalZ), 1); in adaptivemaxpool_loop() local
214 int blocksH = std::max((int)(16L / totalZ), 1); in adaptivemaxgradinput_loop() local
290 int blocksH = std::max((int)(16L / totalZ), 1); in atomicadaptivemaxgradinput_loop() local
H A DAdaptiveAveragePooling3d.cu128 int blocksH = std::max((int)(16L / totalZ), 1); in adaptiveaveragepool_loop() local
225 int blocksH = std::max((int)(16L / totalZ), 1); in adaptiveaveragegradinput_loop() local
320 int blocksH = std::max((int)(16L / totalZ), 1); in atomicadaptiveaveragegradinput_loop() local
H A DAdaptiveAveragePooling.cu580 int blocksH = std::max<int64_t>((int)(16L / sizeD), 1); in adaptive_avg_pool2d_out_cuda_template() local
732 int blocksH = std::max((int)(16L / sizeD), 1); in adaptive_avg_pool2d_backward_out_cuda_template() local