Home
last modified time | relevance | path

Searched defs:pad_l (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReflectionPad.cu37 int64_t pad_l) { in get_index_mapping1d()
62 int64_t pad_l, int64_t pad_t, in get_index_mapping2d()
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()
291 int64_t pad_l = padding[0]; in reflection_pad2d_out_template() local
381 int64_t pad_l = padding[0]; in reflection_pad2d_backward_out_template() local
450 int64_t pad_l = padding[0]; in TORCH_IMPL_FUNC() local
507 int64_t pad_l = padding[0]; in TORCH_IMPL_FUNC() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmaxpooling_op_gpu.cu.cc72 const int pad_l, dtype* __restrict__ top_data, int64* __restrict__ mask, in MaxPoolForwardNCHW()
115 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
148 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNHWC()
235 const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNCHW()
276 const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNHWC()
358 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in operator ()()
378 const int stride_w, const int pad_t, const int pad_l, T* top_data, in operator ()()
429 const int pad_l, const T* top_diff, T* bottom_diff, in operator ()()
H A Dpooling_ops_3d_gpu.cu.cc37 const int pad_l, const dtype* __restrict__ top_diff, in MaxPoolGradBackwardNoMaskNCDHW()
88 const int pad_l, const dtype* __restrict__ top_diff, in MaxPoolGradBackwardNoMaskNDHWC()
143 const int stride_w, const int pad_p, const int pad_t, const int pad_l, in operator ()()
H A Davgpooling_op_gpu.cu.cc88 const int pad_l, T* const bottom_diff, in RunAvePoolBackwardNHWC()
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 Dconv_grad_ops_3d.cc81 const int pad_t, const int pad_l, const int pad_pb, const int pad_b, in Col2im()
130 const int pad_t, const int pad_l, const int pad_pb, const int pad_b, 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()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DReflectionPad.cpp41 auto pad_l = padding[0]; in TORCH_META_FUNC() local
83 auto pad_l = padding[0]; in TORCH_META_FUNC() local
230 int64_t pad_l = padding[0]; in reflection_pad2d_out_template() local
281 int64_t pad_l = padding[0]; in reflection_pad2d_backward_out_template() local
H A DReplicationPadding.cpp34 int64_t pad_l = paddingSize[0]; in TORCH_META_FUNC() local
68 int64_t pad_l = paddingSize[0]; in TORCH_META_FUNC() local
90 int64_t pad_l = paddingSize[0]; in TORCH_META_FUNC() local
187 int pad_l = paddingSize[0]; in replication_pad2d_backward_out_cpu_template() local
H A DPadNd.cpp134 const auto& pad_l = padding[2 * (ndim_padded - i - 1) + 0]; in _pad_circular_symint() local
155 const auto& pad_l = padding[2*i + 0]; in _pad_circular_symint() local
170 const auto& pad_l = padding[2*i + 0]; 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/ComputeLibrary/src/cpu/kernels/pool2d/neon/nchw/
H A Dall.cpp59 float16x4_t read_4_boundary_aware_fp16(int srcw, int srch, int pad_l, int pad_t, int x, int y, cons… in read_4_boundary_aware_fp16()
170 auto read_2_boundary_aware(int srcw, int srch, int pad_l, int pad_t, int x, int y, const T *ptr, T … in read_2_boundary_aware()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_ops.h542 memory::dims* output_dims_mkl_order, memory::dims* pad_l, 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()
8150 const int pad_t, const int pad_l, const int pad_pb, const int pad_b, in Col2im()
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py3734 def pooling_output_shape_pad_lr( argument