| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_convolution_test.cpp | 35 ArrayRef<int64_t> output_padding, in op_convolution_out()
125 int64_t output_padding[] = {0}; in test_dynamic_shape() local
162 int64_t output_padding[1] = {0}; in TEST_F() local
233 int64_t output_padding[] = {0}; in TEST_F() local
295 int64_t output_padding[] = {0}; in TEST_F() local
378 int64_t output_padding[] = {0}; variable
441 int64_t output_padding[] = {0}; variable
487 int64_t output_padding[1] = {0}; in TEST_F() local
535 int64_t output_padding[1] = {0}; in TEST_F() local
572 int64_t output_padding[1] = {2}; in TEST_F() local
[all …]
|
| H A D | op_convolution_backward_test.cpp | 37 IntArrayRef output_padding, in op_convolution_backward_out()
170 int64_t output_padding[2] = {0, 0}; in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qconv_prepack.cpp | 34 torch::List<int64_t> output_padding, in prepack()
202 torch::List<int64_t> output_padding, in prepack()
334 torch::List<int64_t> output_padding, in prepack()
632 torch::List<int64_t> output_padding; in run_conv() local
646 torch::List<int64_t> output_padding, in run_deconv()
659 torch::List<int64_t> output_padding, in _run()
721 const torch::List<int64_t> output_padding({0}); in run_conv() local
731 torch::List<int64_t> output_padding, in run_deconv()
744 torch::List<int64_t> output_padding, in _run()
|
| H A D | conv_serialization.h | 227 auto output_padding = params->output_padding().vec(); in serialize_conv() local
262 auto output_padding = params->output_padding().vec(); in serialize_conv() local
294 torch::List<int64_t> stride, padding, output_padding, dilation; in deserialize_conv() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | ConvPrepack.cpp | 32 torch::List<int64_t> output_padding, in prepack()
140 torch::List<int64_t> output_padding; in run_conv() local
155 torch::List<int64_t> output_padding, in _run()
174 const torch::List<int64_t> output_padding({0}); in run_conv() local
185 torch::List<int64_t> output_padding, in _run()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ConvUtils.h | 253 …ArrayRef<T> padding, ArrayRef<T> output_padding, ArrayRef<T> stride, ArrayRef<T> dilation, T groups in _conv_input_size()
271 …SymIntArrayRef padding, SymIntArrayRef output_padding, SymIntArrayRef stride, SymIntArrayRef dilat… in conv_input_size()
278 …IntArrayRef padding, IntArrayRef output_padding, IntArrayRef stride, IntArrayRef dilation, int64_t… in conv_input_size()
286 …ArrayRef<T> padding, ArrayRef<T> output_padding, IntArrayRef stride, IntArrayRef dilation, int64_t… in _conv_weight_size()
302 …SymIntArrayRef padding, SymIntArrayRef output_padding, IntArrayRef stride, IntArrayRef dilation, i… in conv_weight_size()
309 …IntArrayRef padding, IntArrayRef output_padding, IntArrayRef stride, IntArrayRef dilation, int64_t… in conv_weight_size()
|
| H A D | Convolution.cpp | 290 std::vector<T> output_padding; member
853 SymIntArrayRef output_padding, in complex_convolution()
1031 SymDimVector output_padding(static_cast<size_t>(dim)); in convolution_same() local
1051 SymDimVector output_padding(static_cast<size_t>(dim)); in convolution_same() local
1118 …SymIntArrayRef stride, SymIntArrayRef padding, SymIntArrayRef output_padding, c10::SymInt groups, … in conv_transpose1d_symint()
1137 …SymIntArrayRef stride, SymIntArrayRef padding, SymIntArrayRef output_padding, c10::SymInt groups, … in conv_transpose2d_symint()
1156 …SymIntArrayRef stride, SymIntArrayRef padding, SymIntArrayRef output_padding, c10::SymInt groups, … in conv_transpose3d_symint()
1176 bool transposed, IntArrayRef output_padding, int64_t groups) { in convolution()
1192 bool transposed, IntArrayRef output_padding, int64_t groups) { in convolution_overrideable()
1724 … bool transposed, IntArrayRef output_padding, int64_t groups, std::array<bool, 3> output_mask) { in convolution_backward_overrideable()
[all …]
|
| H A D | NaiveConvolutionTranspose3d.cpp | 185 IntArrayRef output_padding, in slow_conv_transpose3d_out_cpu_template()
399 IntArrayRef output_padding, in slow_conv_transpose3d_backward_out_cpu_template()
603 IntArrayRef output_padding, in slow_conv_transpose3d_acc_grad_parameters_cpu()
825 IntArrayRef output_padding, in slow_conv_transpose3d_out_cpu()
852 IntArrayRef output_padding, in slow_conv_transpose3d_cpu()
881 IntArrayRef output_padding, in slow_conv_transpose3d_backward_cpu()
|
| H A D | NaiveConvolutionTranspose2d.cpp | 252 IntArrayRef output_padding, in slow_conv_transpose2d_out_cpu_template()
397 IntArrayRef output_padding, in slow_conv_transpose2d_backward_out_cpu_template()
595 IntArrayRef output_padding, in slow_conv_transpose2d_acc_grad_parameters_cpu()
812 IntArrayRef output_padding, in slow_conv_transpose2d_backward_cpu()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | NaiveConvolutionTranspose3d.cu | 181 IntArrayRef output_padding, in slow_conv_transpose3d_out_cuda_template()
402 IntArrayRef output_padding, in slow_conv_transpose3d_backward_out_cuda_template()
612 IntArrayRef output_padding, in slow_conv_transpose3d_acc_grad_parameters_cuda()
841 IntArrayRef output_padding, in slow_conv_transpose3d_out_cuda()
868 IntArrayRef output_padding, in slow_conv_transpose3d_cuda()
896 IntArrayRef output_padding, in slow_conv_transpose3d_backward_out_cuda()
949 IntArrayRef output_padding, in slow_conv_transpose3d_backward_cuda()
|
| H A D | NaiveConvolutionTranspose2d.cu | 156 IntArrayRef output_padding, in slow_conv_transpose2d_out_cuda_template()
311 IntArrayRef output_padding, in slow_conv_transpose2d_backward_out_cuda_template()
489 IntArrayRef output_padding, in slow_conv_transpose2d_acc_grad_parameters_cuda_template()
713 IntArrayRef output_padding, in slow_conv_transpose2d_backward_out_cuda()
766 IntArrayRef output_padding, in slow_conv_transpose2d_backward_cuda()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | kernel_ops_util.cpp | 114 IntArrayRef output_padding, in output_padding_is_valid()
205 int64_t output_padding) { in _kernel_output_size_helper()
233 IntArrayRef output_padding) { in calculate_kernel_output_sizes()
327 IntArrayRef output_padding, in check_convolution_args()
413 IntArrayRef output_padding, in get_convolution_out_target_size()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | conv.h | 172 IntArrayRef output_padding, in conv_transpose1d()
217 IntArrayRef output_padding, in conv_transpose2d()
262 IntArrayRef output_padding, in conv_transpose3d()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Conv.cpp | 623 IntArrayRef output_padding, in _mkldnn_convolution_transpose()
719 IntArrayRef output_padding, in mkldnn_convolution_transpose_pointwise()
750 IntArrayRef output_padding, in mkldnn_convolution_transpose_pointwise_meta()
900 IntArrayRef output_padding, in mkldnn_convolution_transpose()
924 IntArrayRef output_padding, in mkldnn_convolution_transpose_backward_input()
967 IntArrayRef output_padding, in mkldnn_convolution_transpose_backward_weights()
1018 …IntArrayRef padding, IntArrayRef output_padding, IntArrayRef stride, IntArrayRef dilation, int64_t… in mkldnn_convolution_transpose_backward()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | conv.cpp | 258 std::vector<int64_t> output_padding = _output_padding( in forward() local
295 std::vector<int64_t> output_padding = _output_padding( in forward() local
332 std::vector<int64_t> output_padding = _output_padding( in forward() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/ |
| H A D | KernelUtils.cpp | 111 const int64_t output_padding) { in calc_transpose_out_size()
124 const utils::ivec2& output_padding) { in calc_transpose_out_sizes_hw()
164 const auto output_padding = make_ivec2_from_list(graph, args[3]); in calc_out_sizes_hw() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesConvolution.cpp | 20 …ef padding, c10::SymIntArrayRef dilation, bool transposed, c10::SymIntArrayRef output_padding, c10… in convolution_batch_rule()
173 c10::SymIntArrayRef output_padding, const c10::SymInt& groups) { in convolution_backward_input_batch_rule()
254 c10::SymIntArrayRef output_padding, const c10::SymInt& groups) { in convolution_backward_weight_batch_rule()
365 c10::SymIntArrayRef output_padding, c10::SymInt groups, std::array<bool, 3> output_mask) { in convolution_backward_plumbing()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Convolution.cpp | 751 IntArrayRef output_padding, in get_conv_transpose_output_size()
773 const IntArrayRef output_padding, in convolution()
934 const auto output_padding = local
1042 std::vector<int64_t>&& output_padding, in create_tconv2d_context()
1089 std::vector<int64_t>&& output_padding, in create_qtconv2d_context()
1139 const auto output_padding = in run_conv2d_context_impl() local
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | conv.py | 67 output_padding, argument
758 output_padding, argument
926 output_padding=0, argument
1049 output_padding=0, argument
1174 output_padding=0, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | Convolution.cpp | 92 const IntArrayRef output_padding, in create_and_run()
175 const IntArrayRef output_padding, in create()
423 std::vector<int64_t> output_padding, in createConv2dTransposeClampPrePackOpContext()
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
| H A D | conv.py | 320 output_padding=0, argument
402 output_padding=0, argument
484 output_padding=0, argument
|
| /aosp_15_r20/external/executorch/backends/vulkan/_passes/ |
| H A D | custom_ops_defs.py | 39 output_padding=0, argument
81 output_padding=0, argument
|
| /aosp_15_r20/external/executorch/backends/vulkan/ |
| H A D | custom_ops_lib.py | 39 output_padding=0, argument
81 output_padding=0, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| H A D | Conv.cpp | 28 std::vector<int64_t> output_padding; member
157 auto output_padding = params.output_padding; in check_shape_forward() local
606 IntArrayRef output_padding, in convolution_backward_overrideable()
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/ |
| H A D | maia_extension.cpp | 37 bool transposed, IntArrayRef output_padding, int64_t groups) { in fake_convolution()
46 IntArrayRef dilation, bool transposed, IntArrayRef output_padding, in fake_convolution_backward()
|