Home
last modified time | relevance | path

Searched defs:output_padding_ (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/
H A DConv.cpp397 IntArrayRef output_padding_, in _convolution_out()
545 IntArrayRef output_padding_, in _convolution()
571 IntArrayRef output_padding_, in convolution_overrideable()
624 IntArrayRef stride_, padding_, dilation_, output_padding_; in convolution_backward_overrideable() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolution.cpp1314 …bool transposed_, SymIntArrayRef output_padding_, c10::SymInt groups_, const at::OptionalSymIntArr… in select_conv_backend()
1471 bool transposed_, IntArrayRef output_padding_, int64_t groups_, in _convolution()
1711 bool transposed_, IntArrayRef output_padding_, int64_t groups_, in _convolution()
1741 bool transposed_, IntArrayRef output_padding_, int64_t groups_, in _convolution_double_backward()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/
H A DOpContext.h121 std::vector<int64_t> output_padding_; variable
H A DCommon.h37 std::array<int64_t, 2> output_padding_; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/
H A Dutils.h177 torch::List<int64_t> output_padding_; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dfbgemm_utils.h199 torch::List<int64_t> output_padding_; member
H A DQnnpackUtils.h300 torch::List<int64_t> output_padding_; member
H A DOnednnUtils.h220 torch::List<int64_t> output_padding_; member
H A Dfbgemm_utils.cpp125 std::array<int, kSpatialDim> output_padding_; in MakeFbgemmConvParam() local
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp1919 c10::SymInt groups_) { in initModule()
1954 std::optional<std::vector<c10::SymInt>> bias_sizes_opt) { in initModule()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesConvolution.cpp129 bool transposed_, IntArrayRef output_padding_, int64_t groups_, in _convolution_decomp()