Home
last modified time | relevance | path

Searched defs:pad_r (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReflectionPad.cu99 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 DReflectionPad.cpp42 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 DReplicationPadding.cpp35 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 DPadNd.cpp135 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 DPool.h62 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 Dconv_grad_input_ops.h80 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Col2im()
H A Deigen_pooling_test.cc401 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 Dconv_grad_ops_3d.cc82 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 Dconv_grad_filter_ops.cc76 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Im2col()
H A Deigen_spatial_convolutions_test.cc1172 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 Dmkl_conv_ops.h543 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 Dquantized.h329 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 Doptimized_ops.h4902 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.py3738 pad_r, argument