Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dvol2col.cuh28 const int pad_h, in vol2col_kernel()
89 const int pad_h, in vol2col()
140 const unsigned pad_h, in vol2im_kernel()
213 const int64_t pad_h, in col2vol()
H A DAveragePool2d.cu37 const int stride_h, const int stride_w, const int pad_h, const int pad_w, in avg_pool2d_out_cuda_frame()
86 const int stride_h, const int stride_w, const int pad_h, const int pad_w, in avg_pool2d_out_cuda_frame_nhwc()
135 const int stride_w, const int pad_h, const int pad_w, in avg_pool2d_backward_out_cuda_frame()
192 const int stride_w, const int pad_h, const int pad_w, in avg_pool2d_backward_out_cuda_frame_nhwc()
H A DDilatedMaxPool2d.cu49 const int stride_w, const int pad_h, const int pad_w, in max_pool_forward_nchw()
181 const int stride_h, const int stride_w, const int pad_h, const int pad_w, in max_pool_backward_nchw()
/aosp_15_r20/external/pytorch/test/
H A Dtest_xnnpack_integration.py111 pad_h, argument
180 pad_h, argument
334 pad_h, argument
449 pad_h, argument
603 pad_h, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dxsmm_conv2d_test.cc39 int pad_h; member
123 int pad_h = param->pad_h; in naive_conv_fp() local
202 int pad_h = pad; in TEST_F() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUnfold3d.cpp239 int64_t pad_h, in Unfold3dCopyKernelImpl()
371 int64_t pad_h, in Unfold3dAccKernelImpl()
452 int64_t pad_h, in Unfold3dCopyCPU()
500 int64_t pad_h, in Unfold3dAccCPU()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DPaddingKernel.cpp153 int64_t pad_h = ndim >= 2 ? p.pads[ndim - 2] : 0; in cpu_padding() local
260 int64_t pad_h = ndim >= 2 ? p.pads[ndim - 2] : 0; in cpu_padding_channels_last() local
334 int64_t pad_h = ndim >= 2 ? p.pads[ndim - 2] : 0; in cpu_padding_backward() local
422 int64_t pad_h = ndim >= 2 ? p.pads[ndim - 2] : 0; in cpu_padding_backward_channels_last() local
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_op.py5156 pad_h, argument
5237 pad_h, argument
5444 pad_h, argument
5527 pad_h, argument
5784 pad_h, argument
5918 pad_h, argument
6410 pad_h, argument
6494 pad_h, argument
6560 self, inputs, stride_d, stride_h, stride_w, pad_d, pad_h, pad_w, o_pad, argument
H A Dtest_quantized_functional.py141 groups, kernel_h, kernel_w, stride_h, stride_w, pad_h, pad_w, dilation, argument
198 stride_d, stride_h, stride_w, pad_d, pad_h, pad_w, dilation, X_scale, argument
/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_qat.py770 pad_h, argument
922 pad_h, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_space_to_depth_pass.cc175 OpBuilder* builder, int32_t pad_h, in GetPadOpForConv2DFilter()
237 int32_t pad_h = block_size - height % block_size; in HandleConv2DFilter() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DConvolutionLayerFixture.h253 …const PaddingInfo pad_h = height_index < pre_pad_layer.size() ? pre_pad_layer[height_index] : Padd… variable
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DNodeFusionMutator.cpp282 …const PaddingInfo pad_h = height_index < padding_list.size() ? padding_list[height_index] : Paddin… in fuse_pad_with_convolution() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqconv.cpp300 const int pad_h = padding_[kSpatialDim - 2]; in apply_impl() local