| /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()
228 int64_t isizeW = input.size(2); in TORCH_IMPL_FUNC() local
269 int64_t isizeW = input_.size(3); in TORCH_IMPL_FUNC() local
354 int64_t isizeW = input.size(2); in TORCH_IMPL_FUNC() local
415 int64_t isizeW = input.size(3); 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()
471 int isizeW = input_.size(3); in adaptive_avg_pool2d_out_cuda_template() local
557 int64_t isizeW = input_.size(-1); in adaptive_avg_pool2d_out_cuda_template() local
623 int isizeW = input.size(3); in adaptive_avg_pool2d_backward_out_cuda_template() local
717 int64_t isizeW = input.size(-1); in adaptive_avg_pool2d_backward_out_cuda_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AdaptiveMaxPooling3d.cpp | 91 int64_t isizeW, in adaptive_max_pool3d_single_out_frame()
169 int64_t isizeW, in adaptive_max_pool3d_out_frame()
200 int64_t isizeW, in adaptive_max_pool3d_backward_single_out_frame()
240 int64_t isizeW, in adaptive_max_pool3d_backward_out_frame()
267 int64_t isizeW = 0; in TORCH_IMPL_FUNC() local
363 int64_t isizeW = 0; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling3d.cpp | 33 int64_t isizeW, in adaptive_avg_pool3d_out_frame()
115 int64_t isizeW = input.size(-1); in adaptive_avg_pool3d_out_cpu_template() local
185 int64_t isizeW, in adaptive_avg_pool3d_backward_out_frame()
243 int64_t isizeW = input.size(-1); 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 | 65 int64_t isizeW, in adaptive_avg_pool_single_out_frame()
189 int64_t isizeW = input.size(-1); in _adaptive_avg_pool() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 1820 int64_t isizeW,
1929 int64_t isizeW,
1964 int64_t isizeW,
|