| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AdaptiveMaxPooling3d.cpp | 53 int64_t osizeW = output_size[2]; in TORCH_META_FUNC() local
94 int64_t osizeW, in adaptive_max_pool3d_single_out_frame()
172 int64_t osizeW, in adaptive_max_pool3d_out_frame()
203 int64_t osizeW) in adaptive_max_pool3d_backward_single_out_frame()
243 int64_t osizeW) in adaptive_max_pool3d_backward_out_frame()
297 int64_t osizeW = output_size[2]; in TORCH_IMPL_FUNC() local
366 int64_t osizeW = 0; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling3d.cpp | 36 int64_t osizeW, in adaptive_avg_pool3d_out_frame()
124 auto osizeW = output_size[2]; in adaptive_avg_pool3d_out_cpu_template() local
188 int64_t osizeW) { in adaptive_avg_pool3d_backward_out_frame()
246 int64_t osizeW = gradOutput.size(-1); in adaptive_avg_pool3d_backward_out_cpu_template() local
|
| H A D | AdaptiveMaxPooling2d.cpp | 42 int64_t osizeW = output_size[1]; 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()
322 int64_t osizeW = output_size[2]; 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()
364 int64_t osizeW = output_size[2]; 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()
220 int64_t osizeW = output_size[1]; in TORCH_IMPL_FUNC() local
357 int64_t osizeW = gradOutput_.size(2); in TORCH_IMPL_FUNC() local
418 int64_t osizeW = gradOutput_.size(3); 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()
479 int osizeW = output_size[1]; in adaptive_avg_pool2d_out_cuda_template() local
564 int64_t osizeW = output_size[1]; in adaptive_avg_pool2d_out_cuda_template() local
633 int osizeW = gradOutput.size(-1); in adaptive_avg_pool2d_backward_out_cuda_template() local
720 int64_t osizeW = gradOutput.size(-1); 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 | 68 int64_t osizeW, in adaptive_avg_pool_single_out_frame()
193 auto osizeW = output_shape[output_shape.size() - 1]; in _adaptive_avg_pool() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 1823 int64_t osizeW,
1931 int64_t osizeW,
1967 int64_t osizeW,
|