| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | Conv_miopen.cpp | 35 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, in miopen_convolution()
42 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, int64_t groups, in miopen_convolution_backward_input()
49 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, int64_t groups, in miopen_convolution_backward_weight()
61 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, int64_t groups, in miopen_convolution_backward()
68 IntArrayRef padding, IntArrayRef output_padding, IntArrayRef stride, IntArrayRef dilation, in miopen_convolution_transpose()
75 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, in miopen_convolution_transpose_backward_input()
82 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, int64_t groups, in miopen_convolution_transpose_backward_weight()
89 …IntArrayRef padding, IntArrayRef output_padding, IntArrayRef stride, IntArrayRef dilation, int64_t… in miopen_convolution_transpose_backward()
96 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, in miopen_depthwise_convolution()
103 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, int64_t groups, in miopen_depthwise_convolution_backward_input()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | ConvPlaceholders.cpp | 33 IntArrayRef dilation, in cudnn_convolution()
46 IntArrayRef dilation, in cudnn_convolution_out()
61 IntArrayRef dilation, in cudnn_convolution_backward_input()
76 IntArrayRef dilation, in cudnn_convolution_backward_weight()
91 IntArrayRef dilation, in cudnn_convolution_backward()
106 IntArrayRef dilation, in cudnn_convolution_transpose()
119 IntArrayRef dilation, in cudnn_convolution_transpose_backward_input()
134 IntArrayRef dilation, in cudnn_convolution_transpose_backward_weight()
150 IntArrayRef dilation, in cudnn_convolution_transpose_backward()
166 IntArrayRef dilation, in raw_cudnn_convolution_forward_out()
[all …]
|
| H A D | ConvShared.cpp | 109 IntArrayRef dilation, in setConvolutionParams()
212 IntArrayRef dilation, in cudnn_convolution_forward_out()
245 IntArrayRef dilation, in cudnn_convolution()
281 IntArrayRef dilation, in cudnn_convolution_out()
314 IntArrayRef dilation, in cudnn_convolution_transpose_backward_input()
372 IntArrayRef dilation, in cudnn_convolution_backward_input()
414 IntArrayRef dilation, in cudnn_convolution_transpose_forward()
447 IntArrayRef dilation, in cudnn_convolution_backward_input()
474 IntArrayRef dilation, in cudnn_convolution_transpose()
509 IntArrayRef dilation, in cudnn_convolution_backward_weight()
[all …]
|
| H A D | Conv_v8.cpp | 121 IntArrayRef dilation, in getConvDescriptor()
438 const IntArrayRef dilation) { in build_opgraph()
472 const IntArrayRef dilation) { in build_opgraph_fused()
705 const IntArrayRef dilation, in get_plans_from_find()
754 const IntArrayRef dilation, in get_plans_from_find_fused()
810 const IntArrayRef dilation, in get_configs_from_heuristics()
841 const IntArrayRef dilation, in get_configs_from_heuristics_fused()
981 const IntArrayRef dilation, in run_single_conv()
1079 IntArrayRef dilation, in run_fused_conv()
1178 const IntArrayRef dilation, in raw_cudnn_convolution_forward_out()
[all …]
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_convolution_test.cpp | 33 ArrayRef<int64_t> dilation, in op_convolution_out()
124 int64_t dilation[] = {1}; in test_dynamic_shape() local
161 int64_t dilation[1] = {1}; in TEST_F() local
232 int64_t dilation[] = {1}; in TEST_F() local
294 int64_t dilation[] = {1}; in TEST_F() local
377 int64_t dilation[] = {1, 1}; variable
440 int64_t dilation[] = {1, 1}; variable
486 int64_t dilation[1] = {1}; in TEST_F() local
533 int64_t dilation[1] = {1}; in TEST_F() local
570 int64_t dilation[1] = {5}; in TEST_F() local
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Convolution.cpp | 60 const IntArrayRef dilation, in determine_method()
289 const IntArrayRef dilation, in get_shader()
372 const IntArrayRef dilation, in record_op()
445 const IntArrayRef dilation, in record_quantized_op()
575 const IntArrayRef dilation, in compute_overlay_region()
662 const IntArrayRef dilation, in available()
753 IntArrayRef dilation = IntArrayRef()) { in get_conv_transpose_output_size()
771 const IntArrayRef dilation, in convolution()
826 IntArrayRef dilation, in run_conv1d_context_impl()
873 int32_t dilation; in run_conv1d_context_impl() member
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Conv.cpp | 26 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, int64_t groups) { in mkldnn_convolution()
51 const IntArrayRef& dilation, in check_shape_forward()
172 IntArrayRef dilation, in _mkldnn_convolution_out()
219 IntArrayRef dilation, in _mkldnn_convolution()
284 IntArrayRef dilation, in mkldnn_convolution()
305 IntArrayRef dilation, in mkldnn_convolution_pointwise()
342 IntArrayRef dilation, in mkldnn_convolution_pointwise_binary()
506 IntArrayRef dilation, in mkldnn_convolution_pointwise_binary_()
625 IntArrayRef dilation, in _mkldnn_convolution_transpose()
721 IntArrayRef dilation, in mkldnn_convolution_transpose_pointwise()
[all …]
|
| H A D | Pooling.cpp | 37 IntArrayRef dilation, in mkldnn_max_pool2d()
47 IntArrayRef dilation, in mkldnn_max_pool3d()
119 IntArrayRef dilation, in mkldnn_max_pool2d_backward()
131 IntArrayRef dilation, in mkldnn_max_pool3d_backward()
206 IntArrayRef dilation, in _mkldnn_pooling()
297 IntArrayRef dilation, in _mkldnn_pooling_backward()
371 IntArrayRef dilation, in mkldnn_max_pool2d()
395 IntArrayRef dilation, in mkldnn_max_pool3d()
539 IntArrayRef dilation, in mkldnn_max_pool2d_backward()
560 IntArrayRef dilation, in mkldnn_max_pool3d_backward()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/heuristics/dwc_native/ |
| H A D | ClDWCNativeDefaultConfigBifrost.cpp | 38 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure_f32()
108 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure_f16()
201 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure()
230 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure_G71_f32()
236 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure_G71_f16()
242 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure_G7x_f32()
248 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure_G7x_f16()
254 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure_G7x_u8()
|
| H A D | ClDWCNativeDefaultConfigValhall.cpp | 41 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure()
71 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure_G78_f32()
133 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure_G78_f16()
212 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure_G78_u8()
237 …t ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info, const Size2D &dilation, in configure_G77_f16()
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/_expanded_weights/ |
| H A D | conv_utils.py | 44 dilation=1, argument
56 def conv_input_for_string_padding(func, padding_style, input, dilation, kernel_size): argument
66 def int_padding_for_string_padding(func, padding_style, dilation, kernel_size): argument
84 def conv_padding_for_same(dilation, kernel_size): argument
123 def calc_total_padding(func, was_same, padding, dilation, kernel_size): argument
214 dilation, argument
264 dilation, argument
300 dilation, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Convolution.cpp | 261 const IntArrayRef dilation, in xnnpack_use_convolution2d()
273 const SymIntArrayRef dilation, in xnnpack_use_convolution2d()
288 std::vector<T> dilation; member
663 const auto& dilation = params.dilation; in check_shape_forward() local
851 SymIntArrayRef dilation, in complex_convolution()
889 c10::SymIntArrayRef dilation, in complex_convolution_mode()
917 SymIntArrayRef stride, SymIntArrayRef padding, SymIntArrayRef dilation, c10::SymInt groups) { in conv1d_symint()
942 SymIntArrayRef stride, SymIntArrayRef padding, SymIntArrayRef dilation, c10::SymInt groups) { in conv2d_symint()
967 SymIntArrayRef stride, SymIntArrayRef padding, SymIntArrayRef dilation, c10::SymInt groups) { in conv3d_symint()
993 SymIntArrayRef stride, SymIntArrayRef dilation, const c10::SymInt& groups) { in convolution_same()
[all …]
|
| H A D | DilatedMaxPool3d.cpp | 28 IntArrayRef dilation, in max_pool3d_with_indices_out_cpu_template()
120 IntArrayRef dilation, in max_pool3d_with_indices_backward_out_cpu_template()
204 IntArrayRef dilation, in max_pool3d_with_indices_out_cpu()
226 IntArrayRef dilation, in max_pool3d_with_indices_cpu()
255 IntArrayRef dilation, in max_pool3d_with_indices_backward_out_cpu()
279 IntArrayRef dilation, in max_pool3d_with_indices_backward_cpu()
|
| H A D | ConvUtils.h | 194 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, int64_t groups) in convolution_shape_check()
253 …ArrayRef<T> padding, ArrayRef<T> output_padding, ArrayRef<T> stride, ArrayRef<T> dilation, T groups in _conv_input_size()
271 …f padding, SymIntArrayRef output_padding, SymIntArrayRef stride, SymIntArrayRef dilation, c10::Sym… 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()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qconv_prepack.cpp | 35 torch::List<int64_t> dilation, in prepack()
203 torch::List<int64_t> dilation, in prepack()
335 torch::List<int64_t> dilation, in prepack()
500 torch::List<int64_t> dilation, in _qconv_prepack_onednn()
630 torch::List<int64_t> dilation, in run_conv()
647 torch::List<int64_t> dilation, in run_deconv()
660 torch::List<int64_t> dilation, in _run()
719 torch::List<int64_t> dilation, in run_conv()
732 torch::List<int64_t> dilation, in run_deconv()
745 torch::List<int64_t> dilation, in _run()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | grad.py | 14 dilation=1, argument
64 dilation=1, argument
112 dilation=1, argument
162 dilation=1, argument
210 dilation=1, argument
260 dilation=1, argument
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | ConvolutionLayerFixture.h | 68 … const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups) in configure_conv_function()
79 … const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups) in configure_conv_function()
366 …ensorShape bias_shape, TensorShape output_shape, PadStrideInfo info, Size2D dilation, bool reshape… in setup()
394 …ensorShape bias_shape, TensorShape output_shape, PadStrideInfo info, Size2D dilation, bool reshape… in setup()
407 …ensorShape bias_shape, TensorShape output_shape, PadStrideInfo info, Size2D dilation, bool reshape… in setup()
518 …ensorShape bias_shape, TensorShape output_shape, PadStrideInfo info, Size2D dilation, DataLayout d… in setup()
565 const Size2D &dilation) in compute_target()
594 const Size2D &dilation) in compute_reference()
623 const Size2D &dilation) in configure_and_execute_kernel()
666 const Size2D &dilation) in configure_and_execute_kernel()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | conv.h | 32 ExpandingArray<1> dilation, in conv1d()
78 ExpandingArray<2> dilation, in conv2d()
124 ExpandingArray<3> dilation, in conv3d()
174 IntArrayRef dilation) { in conv_transpose1d()
219 IntArrayRef dilation) { in conv_transpose2d()
264 IntArrayRef dilation) { in conv_transpose3d()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | Convolution.cpp | 32 const IntArrayRef dilation, in available()
94 const IntArrayRef dilation, in create_and_run()
177 const IntArrayRef dilation, in create()
402 std::vector<int64_t> dilation, in createConv2dClampPrePackOpContext()
424 std::vector<int64_t> dilation, in createConv2dTransposeClampPrePackOpContext()
476 const IntArrayRef dilation, in use_convolution2d()
498 const IntArrayRef dilation, in convolution2d()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | kernel_ops_util.cpp | 108 bool dilation_is_valid(IntArrayRef dilation, size_t kernel_ndim) { in dilation_is_valid()
116 IntArrayRef dilation, in output_padding_is_valid()
202 int64_t dilation, in _kernel_output_size_helper()
229 IntArrayRef dilation, in calculate_kernel_output_sizes()
325 IntArrayRef dilation, in check_convolution_args()
411 IntArrayRef dilation, in get_convolution_out_target_size()
470 IntArrayRef dilation, in check_max_pool2d_with_indices_args()
504 IntArrayRef dilation, in get_max_pool2d_with_indices_out_target_size()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset13.py | 887 dilation, argument
913 dilation, argument
939 dilation, argument
965 dilation, argument
990 dilation, argument
1015 dilation, argument
1041 dilation, argument
1069 dilation, argument
1097 dilation, argument
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | DepthwiseConvolutionLayer.cpp | 55 … unsigned int depth_multiplier, const Size2D &dilation, const QuantizationInfo &out_quant_info) in depthwise_convolution_fp()
139 …const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation, const Quant… in depthwise_convolution_quantized()
239 …const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation, const Quant… in depthwise_convolution()
246 …const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation, const Quant… in depthwise_convolution()
253 …const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation, const Quant… in depthwise_convolution()
260 …const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation, const Quant… in depthwise_convolution()
267 …const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation, const Quant… in depthwise_convolution()
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/qat/modules/ |
| H A D | conv_fused.py | 51 dilation, argument
467 dilation=1, argument
538 dilation=1, argument
600 dilation=1, argument
664 dilation=1, argument
735 dilation=1, argument
797 dilation=1, argument
861 dilation=1, argument
931 dilation=1, argument
995 dilation=1, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | ConvPrepack.cpp | 33 torch::List<int64_t> dilation, in prepack()
138 torch::List<int64_t> dilation, in run_conv()
156 torch::List<int64_t> dilation, in _run()
172 torch::List<int64_t> dilation, in run_conv()
186 torch::List<int64_t> dilation, in _run()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/ |
| H A D | KernelUtils.cpp | 35 const ValueRef dilation) { in create_kernel2d_params()
62 const int64_t dilation, in calc_out_size()
80 const utils::ivec2& dilation, in calc_out_sizes_hw()
110 const int64_t dilation, in calc_transpose_out_size()
123 const utils::ivec2& dilation, in calc_transpose_out_sizes_hw()
|