/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | ReflectionPad.cu | 99 int64_t pad_l, int64_t pad_r) { in reflection_pad1d_out_kernel() 113 int64_t pad_l, int64_t pad_r) { in reflection_pad1d_backward_out_kernel() 128 int pad_t, int pad_b, int pad_l, int pad_r, int y_shift, int z_shift, int nplane) { in reflection_pad2d_out_kernel() 148 int pad_t, int pad_b, int pad_l, int pad_r, int y_shift, int z_shift, int nplane) { in reflection_pad2d_backward_out_kernel() 292 int64_t pad_r = padding[1]; in reflection_pad2d_out_template() local 382 int64_t pad_r = padding[1]; in reflection_pad2d_backward_out_template() local 451 int64_t pad_r = padding[1]; in TORCH_IMPL_FUNC() local 508 int64_t pad_r = padding[1]; in TORCH_IMPL_FUNC() local
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | ReflectionPad.cpp | 42 auto pad_r = padding[1]; in TORCH_META_FUNC() local 84 auto pad_r = padding[1]; in TORCH_META_FUNC() local 231 int64_t pad_r = padding[1]; in reflection_pad2d_out_template() local 282 int64_t pad_r = padding[1]; in reflection_pad2d_backward_out_template() local
|
H A D | ReplicationPadding.cpp | 35 int64_t pad_r = paddingSize[1]; in TORCH_META_FUNC() local 69 int64_t pad_r = paddingSize[1]; in TORCH_META_FUNC() local 91 int64_t pad_r = paddingSize[1]; in TORCH_META_FUNC() local 188 int pad_r = paddingSize[1]; in replication_pad2d_backward_out_cpu_template() local
|
H A D | PadNd.cpp | 135 const auto& pad_r = padding[2 * (ndim_padded - i - 1) + 1]; in _pad_circular_symint() local 156 const auto& pad_r = padding[2*i + 1]; in _pad_circular_symint() local 171 const auto& pad_r = padding[2*i + 1]; in _pad_circular_symint() local
|
H A D | Pool.h | 62 T inputSize, T kernelSize, T pad_l, T pad_r, T stride, T dilation, in pooling_output_shape_pad_lr()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | conv_grad_input_ops.h | 80 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Col2im()
|
H A D | eigen_pooling_test.cc | 401 const int pad_r = output_rows - input_rows + patch_rows - 1; in TEST() local 479 const int pad_r = output_rows - input_rows + patch_rows - 1; in TEST() local
|
H A D | conv_grad_ops_3d.cc | 82 const int pad_r, const int stride_p, const int stride_h, in Col2im() 131 const int pad_r, const int stride_p, const int stride_h, in Im2col()
|
H A D | conv_grad_filter_ops.cc | 76 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Im2col()
|
H A D | eigen_spatial_convolutions_test.cc | 1172 const int pad_r = (out_height - 1) * stride - in_rows + kern_height; in TEST() local 1243 const int pad_r = (out_height - 1) * stride - in_rows + kern_height; in TEST() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
H A D | mkl_conv_ops.h | 543 memory::dims* pad_r, bool is_grouped_convolution, bool is_depthwise) { in GetOutputAndPadSizeInMklOrder()
|
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/pool2d/neon/ |
H A D | quantized.h | 329 auto load16_boundary_aware(int srcw, int srch, int pad_l, int pad_r, int pad_t, int pad_b, int x, i… in load16_boundary_aware()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
H A D | optimized_ops.h | 4902 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Col2im() 8151 const int pad_r, const int stride_p, const int stride_h, in Col2im()
|
/aosp_15_r20/external/pytorch/torch/ |
H A D | _meta_registrations.py | 3738 pad_r, argument
|