| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AdaptiveMaxPooling2d.cu | 47 int isizeH, int isizeW, in adaptivemaxpool()
113 int isizeH, int isizeW, in adaptivemaxgradinput()
161 int isizeH, int isizeW, int osizeH, int osizeW in atomicadaptivemaxgradinput()
227 int64_t isizeH = input.size(1); in TORCH_IMPL_FUNC() local
268 int64_t isizeH = input_.size(2); in TORCH_IMPL_FUNC() local
353 int64_t isizeH = input.size(1); in TORCH_IMPL_FUNC() local
414 int64_t isizeH = input.size(2); in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveMaxPooling3d.cu | 48 int isizeT, int isizeH, int isizeW, in adaptivemaxpool()
128 int isizeT, int isizeH, int isizeW, in adaptivemaxpool_loop()
162 int isizeT, int isizeH, int isizeW, in adaptivemaxgradinput()
208 int isizeT, int isizeH, int isizeW, in adaptivemaxgradinput_loop()
238 int isizeT, int isizeH, int isizeW, in atomicadaptivemaxgradinput()
284 int isizeT, int isizeH, int isizeW, in atomicadaptivemaxgradinput_loop()
324 int64_t sizeD, isizeT, isizeH, isizeW; in TORCH_IMPL_FUNC() local
406 int64_t sizeD, isizeT, isizeH, isizeW; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling3d.cu | 54 int isizeT, int isizeH, int isizeW, in adaptiveaveragepool()
122 int isizeT, int isizeH, int isizeW, in adaptiveaveragepool_loop()
155 int isizeT, int isizeH, int isizeW, in adaptiveaveragegradinput()
220 int isizeT, int isizeH, int isizeW, in adaptiveaveragegradinput_loop()
253 int isizeT, int isizeH, int isizeW, in atomicadaptiveaveragegradinput()
316 int isizeT, int isizeH, int isizeW, in atomicadaptiveaveragegradinput_loop()
366 int64_t sizeD, isizeT, isizeH, isizeW; in adaptive_avg_pool3d_out_cuda_template() local
445 int64_t sizeD, isizeT, isizeH, isizeW; in adaptive_avg_pool3d_backward_out_cuda_template() local
|
| H A D | AdaptiveAveragePooling.cu | 54 int isizeH, int isizeW, 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()
470 int isizeH = input_.size(2); in adaptive_avg_pool2d_out_cuda_template() local
556 int64_t isizeH = input_.size(-2); in adaptive_avg_pool2d_out_cuda_template() local
622 int isizeH = input.size(2); in adaptive_avg_pool2d_backward_out_cuda_template() local
716 int64_t isizeH = input.size(-2); in adaptive_avg_pool2d_backward_out_cuda_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AdaptiveMaxPooling3d.cpp | 90 int64_t isizeH, in adaptive_max_pool3d_single_out_frame()
168 int64_t isizeH, in adaptive_max_pool3d_out_frame()
199 int64_t isizeH, in adaptive_max_pool3d_backward_single_out_frame()
239 int64_t isizeH, in adaptive_max_pool3d_backward_out_frame()
266 int64_t isizeH = 0; in TORCH_IMPL_FUNC() local
362 int64_t isizeH = 0; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling3d.cpp | 32 int64_t isizeH, in adaptive_avg_pool3d_out_frame()
114 int64_t isizeH = input.size(-2); in adaptive_avg_pool3d_out_cpu_template() local
184 int64_t isizeH, in adaptive_avg_pool3d_backward_out_frame()
242 int64_t isizeH = input.size(-2); in adaptive_avg_pool3d_backward_out_cpu_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | AdaptiveAveragePooling.cpp | 64 int64_t isizeH, in adaptive_avg_pool_single_out_frame()
188 int64_t isizeH = input.size(-2); in _adaptive_avg_pool() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 1819 int64_t isizeH,
1928 int64_t isizeH,
1963 int64_t isizeH,
|