| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | AvgPoolKernel.cpp | 22 int64_t padW, int64_t padH, in cpu_avg_pool2d()
109 int64_t padW, int64_t padH, in cpu_avg_pool2d_channels_last()
223 int64_t padW, int64_t padH, in cpu_avg_pool2d_channels_last()
354 int padW, int padH, in cpu_avg_pool2d_backward()
422 int padW, int padH, in cpu_avg_pool2d_backward_channels_last()
501 int64_t padW, int64_t padH, in avg_pool2d_kernel_impl()
527 int padW, int padH, in avg_pool2d_backward_kernel_impl()
555 int64_t padW, int64_t padH, int64_t padD, in cpu_avg_pool3d()
651 int64_t padW, int64_t padH, int64_t padD, in cpu_avg_pool3d_channels_last()
774 int64_t padW, int64_t padH, int64_t padD, in cpu_avg_pool3d_channels_last()
[all …]
|
| H A D | Unfold2d.cpp | 44 int64_t padH, in unfolded2d_acc()
129 int64_t padH, in unfolded2d_acc_channels_last()
186 int64_t padH, in unfolded2d_acc_kernel()
237 int64_t padH, in unfolded2d_copy()
344 int64_t padH, in unfolded2d_copy_channels_last()
401 int64_t padH, in unfolded2d_copy_kernel()
|
| H A D | MaxPoolKernel.cpp | 252 int padH = padWHD[1]; in cpu_max_pool() local
381 int padH = padWHD[1]; in cpu_max_pool_channels_last() local
605 int padW, int padH, in max_pool2d_kernel_impl()
631 int padW, int padH, int padD, in max_pool3d_kernel_impl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AveragePool3d.cpp | 47 const int padH = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[1]); in TORCH_META_FUNC() local
116 const int padH = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[1]); in TORCH_META_FUNC() local
174 int padH, in avg_pool3d_out_frame()
275 const int padH = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[1]); in TORCH_IMPL_FUNC() local
360 int padH, in avg_pool3d_backward_out_frame()
448 const int padH = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[1]); in TORCH_IMPL_FUNC() local
|
| H A D | AveragePool2d.cpp | 38 const int64_t padH = padding[0]; in TORCH_PRECOMPUTE_META_FUNC() local
120 const int padH = safe_downcast<int, int64_t>(padding[0]); in TORCH_META_FUNC() local
196 const int padH = safe_downcast<int, int64_t>(padding[0]); in TORCH_IMPL_FUNC() local
|
| H A D | DilatedMaxPool2d.cpp | 41 const int padH = safe_downcast<int, int64_t>(padding[0]); in TORCH_META_FUNC() local
114 const int padH = safe_downcast<int, int64_t>(padding[0]); in TORCH_META_FUNC() local
180 const int padH = safe_downcast<int, int64_t>(padding[0]); in TORCH_IMPL_FUNC() local
|
| H A D | Pool.h | 123 …int64_t kH, int64_t kW, int64_t dH, int64_t dW, int64_t padH, int64_t padW, int64_t dilationH, int… in pool2d_shape_check()
174 int kH, int kW, int dH, int dW, int padH, int padW, int dilationH, int dilationW, in max_pool2d_backward_shape_check()
202 int kH, int kW, int dH, int dW, int padH, int padW, in avg_pool2d_backward_shape_check()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AveragePool3d.cu | 41 int padT, int padH, int padW, in avg_pool3d_cuda_update_output()
110 int padT, int padH, int padW, in avg_pool3d_cuda_update_output()
221 int padT, int padH, int padW, in avg_pool3d_cuda_update_grad_input_atomic()
280 int padT, int padH, int padW, in avg_pool3d_cuda_update_grad_input()
373 const int padH = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[1]); in TORCH_IMPL_FUNC() local
479 const int padH = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[1]); in TORCH_IMPL_FUNC() local
|
| H A D | ConvolutionMM2d.cu | 29 int64_t padH, int64_t padW, in slow_conv2d_shape_check()
132 int64_t padH, int64_t padW) { in slow_conv2d_forward()
224 int padH, int padW) { in slow_conv2d_backward()
298 int64_t padH, int64_t padW) { in slow_conv2d_grad_weight()
|
| H A D | DepthwiseConv2d.cu | 367 const int padW, const int padH, in conv_depthwise2d_forward_out()
460 const int padW, const int padH, in conv_depthwise2d_backward_out()
591 const int padW, const int padH, in conv_depthwise2d_grad_weight_out()
|
| H A D | AveragePool2d.cu | 266 const int padH = safe_downcast<int, int64_t>(padH_); in TORCH_IMPL_FUNC() local
382 const int padH = safe_downcast<int, int64_t>(padding[0]); in TORCH_IMPL_FUNC() local
|
| H A D | DilatedMaxPool2d.cu | 321 const int padH = safe_downcast<int, int64_t>(padding[0]); in TORCH_IMPL_FUNC() local
447 const int padH = safe_downcast<int, int64_t>(padding[0]); in TORCH_IMPL_FUNC() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | AveragePool2d.cpp | 48 int padH, in avg_pool2d_out_frame()
151 const int padH = safe_downcast<int, int64_t>(padding[0]); in get_padding() local
164 int padH, in get_output_shape()
|
| H A D | AveragePool3d.cpp | 59 const int padH = in get_padding() local
75 int padH, in get_output_shape()
|
| H A D | Pooling.cpp | 537 int64_t padH = padding[0]; in qnnpack_maxpool2d() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | Pooling.cpp | 140 int64_t padH = padding[0]; in quantized_max_pool2d_cudnn() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 2013 int padH,
2126 int padH,
2173 int padH,
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 2512 padH, argument
3782 padH, argument
|