| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AdaptiveMaxPooling3d.cpp | 52 int64_t osizeH = output_size[1]; in TORCH_META_FUNC() local
93 int64_t osizeH, in adaptive_max_pool3d_single_out_frame()
171 int64_t osizeH, in adaptive_max_pool3d_out_frame()
202 int64_t osizeH, in adaptive_max_pool3d_backward_single_out_frame()
242 int64_t osizeH, in adaptive_max_pool3d_backward_out_frame()
296 int64_t osizeH = output_size[1]; in TORCH_IMPL_FUNC() local
365 int64_t osizeH = 0; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling3d.cpp | 35 int64_t osizeH, in adaptive_avg_pool3d_out_frame()
123 auto osizeH = output_size[1]; in adaptive_avg_pool3d_out_cpu_template() local
187 int64_t osizeH, in adaptive_avg_pool3d_backward_out_frame()
245 int64_t osizeH = gradOutput.size(-2); in adaptive_avg_pool3d_backward_out_cpu_template() local
|
| H A D | AdaptiveMaxPooling2d.cpp | 41 int64_t osizeH = output_size[0]; in TORCH_META_FUNC() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AdaptiveMaxPooling3d.cu | 49 int osizeT, int osizeH, int osizeW, in adaptivemaxpool()
129 int osizeT, int osizeH, int osizeW, in adaptivemaxpool_loop()
163 int osizeT, int osizeH, int osizeW, in adaptivemaxgradinput()
209 int osizeT, int osizeH, int osizeW) in adaptivemaxgradinput_loop()
239 int osizeT, int osizeH, int osizeW, in atomicadaptivemaxgradinput()
285 int osizeT, int osizeH, int osizeW) in atomicadaptivemaxgradinput_loop()
321 int64_t osizeH = output_size[1]; in TORCH_IMPL_FUNC() local
407 int64_t osizeT, osizeH, osizeW; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling3d.cu | 55 int osizeT, int osizeH, int osizeW, in adaptiveaveragepool()
123 int osizeT, int osizeH, int osizeW, in adaptiveaveragepool_loop()
156 int osizeT, int osizeH, int osizeW, in adaptiveaveragegradinput()
221 int osizeT, int osizeH, int osizeW) { in adaptiveaveragegradinput_loop()
254 int osizeT, int osizeH, int osizeW, in atomicadaptiveaveragegradinput()
317 int osizeT, int osizeH, int osizeW) { in atomicadaptiveaveragegradinput_loop()
363 int64_t osizeH = output_size[1]; in adaptive_avg_pool3d_out_cuda_template() local
446 int64_t osizeT, osizeH, osizeW; in adaptive_avg_pool3d_backward_out_cuda_template() local
|
| H A D | AdaptiveMaxPooling2d.cu | 48 int osizeH, int osizeW, in adaptivemaxpool()
114 int osizeH, int osizeW) in adaptivemaxgradinput()
161 int isizeH, int isizeW, int osizeH, int osizeW in atomicadaptivemaxgradinput()
219 int64_t osizeH = output_size[0]; in TORCH_IMPL_FUNC() local
356 int64_t osizeH = gradOutput_.size(1); in TORCH_IMPL_FUNC() local
417 int64_t osizeH = gradOutput_.size(2); in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling.cu | 55 int osizeH, int osizeW, in adaptive_average_pool()
115 int isizeH, int isizeW, int osizeH, int osizeW in adaptive_average_gradinput()
171 int isizeH, int isizeW, int osizeH, int osizeW in atomic_adaptive_average_gradinput()
478 int osizeH = output_size[0]; in adaptive_avg_pool2d_out_cuda_template() local
563 int64_t osizeH = output_size[0]; in adaptive_avg_pool2d_out_cuda_template() local
632 int osizeH = gradOutput.size(-2); in adaptive_avg_pool2d_backward_out_cuda_template() local
719 int64_t osizeH = gradOutput.size(-2); in adaptive_avg_pool2d_backward_out_cuda_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | AdaptiveAveragePooling.cpp | 67 int64_t osizeH, in adaptive_avg_pool_single_out_frame()
192 auto osizeH = output_shape[output_shape.size() - 2]; in _adaptive_avg_pool() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 1822 int64_t osizeH,
1930 int64_t osizeH,
1966 int64_t osizeH,
|