| /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 | 45 int64_t padW, in unfolded2d_acc()
130 int64_t padW, in unfolded2d_acc_channels_last()
187 int64_t padW, in unfolded2d_acc_kernel()
238 int64_t padW, in unfolded2d_copy()
345 int64_t padW, in unfolded2d_copy_channels_last()
402 int64_t padW, in unfolded2d_copy_kernel()
|
| H A D | MaxPoolKernel.cpp | 251 int padW = padWHD[0]; in cpu_max_pool() local
380 int padW = padWHD[0]; 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 | 48 const int padW = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[2]); in TORCH_META_FUNC() local
117 const int padW = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[2]); in TORCH_META_FUNC() local
173 int padW, in avg_pool3d_out_frame()
276 const int padW = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[2]); in TORCH_IMPL_FUNC() local
359 int padW, in avg_pool3d_backward_out_frame()
449 const int padW = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[2]); in TORCH_IMPL_FUNC() local
|
| H A D | AveragePool2d.cpp | 39 const int64_t padW = padding.size() == 1 ? padH : padding[1]; in TORCH_PRECOMPUTE_META_FUNC() local
121 const int padW = padding.size() == 1 ? padH : safe_downcast<int, int64_t>(padding[1]); in TORCH_META_FUNC() local
197 const int padW = padding.size() == 1 ? padH : safe_downcast<int, int64_t>(padding[1]); in TORCH_IMPL_FUNC() local
|
| H A D | DilatedMaxPool2d.cpp | 42 const int padW = padding.size() == 1 ? padH : safe_downcast<int, int64_t>(padding[1]); in TORCH_META_FUNC() local
115 const int padW = padding.size() == 1 ? padH : safe_downcast<int, int64_t>(padding[1]); in TORCH_META_FUNC() local
181 const int padW = padding.size() == 1 ? padH : safe_downcast<int, int64_t>(padding[1]); 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()
374 const int padW = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[2]); in TORCH_IMPL_FUNC() local
480 const int padW = padding.size() == 1 ? padT : safe_downcast<int, int64_t>(padding[2]); 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 | 267 const int padW = safe_downcast<int, int64_t>(padW_); in TORCH_IMPL_FUNC() local
383 const int padW = padding.size() == 1 ? padH : safe_downcast<int, int64_t>(padding[1]); in TORCH_IMPL_FUNC() local
|
| H A D | DilatedMaxPool2d.cu | 322 const int padW = padding.size() == 1 ? padH : safe_downcast<int, int64_t>(padding[1]); in TORCH_IMPL_FUNC() local
448 const int padW = padding.size() == 1 ? padH : safe_downcast<int, int64_t>(padding[1]); in TORCH_IMPL_FUNC() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | AveragePool2d.cpp | 47 int padW, in avg_pool2d_out_frame()
152 const int padW = in get_padding() local
163 int padW, in get_output_shape()
|
| H A D | AveragePool3d.cpp | 61 const int padW = in get_padding() local
74 int padW, in get_output_shape()
|
| H A D | Pooling.cpp | 538 int64_t padW = padding[1]; in qnnpack_maxpool2d() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | Pooling.cpp | 141 int64_t padW = padding[1]; in quantized_max_pool2d_cudnn() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 2012 int padW,
2125 int padW,
2172 int padW,
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 2513 padW, argument
3783 padW, argument
|