Home
last modified time | relevance | path

Searched defs:dilation (Results 1 – 25 of 230) sorted by relevance

12345678910

/external/pytorch/torch/nn/utils/_expanded_weights/
Dconv_utils.py44 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
/external/ComputeLibrary/src/runtime/heuristics/dwc_native/
DClDWCNativeDefaultConfigBifrost.cpp38 …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()
DClDWCNativeDefaultConfigValhall.cpp41 …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()
/external/pytorch/aten/src/ATen/native/cudnn/
DConvPlaceholders.cpp33 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 …]
DConvShared.cpp109 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 …]
/external/pytorch/torch/nn/
Dgrad.py14 dilation=1, argument
64 dilation=1, argument
112 dilation=1, argument
162 dilation=1, argument
210 dilation=1, argument
260 dilation=1, argument
/external/pytorch/torch/csrc/api/include/torch/nn/functional/
Dconv.h32 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()
Dfold.h15 ExpandingArray<2> dilation, in fold()
62 ExpandingArray<2> dilation, in unfold()
/external/ComputeLibrary/src/dynamic_fusion/sketch/attributes/
DConv2dAttributes.cpp51 Conv2dAttributes &Conv2dAttributes::dilation(const Size2D &dilation) in dilation() function in arm_compute::experimental::dynamic_fusion::Conv2dAttributes
56 Size2D Conv2dAttributes::dilation() const in dilation() function in arm_compute::experimental::dynamic_fusion::Conv2dAttributes
DDepthwiseConv2dAttributes.cpp51 DepthwiseConv2dAttributes &DepthwiseConv2dAttributes::dilation(const Size2D &dilation) in dilation() function in arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes
56 Size2D DepthwiseConv2dAttributes::dilation() const in dilation() function in arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes
/external/pytorch/aten/src/ATen/native/quantized/cudnn/
DConvPrepack.cpp33 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()
/external/executorch/kernels/test/
Dop_convolution_test.cpp33 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 …]
Dop_max_pool2d_with_indices_test.cpp28 exec_aten::ArrayRef<int64_t> dilation, in op_max_pool2d_with_indices_out()
80 exec_aten::ArrayRef<int64_t> dilation = in TEST_F() local
176 exec_aten::ArrayRef<int64_t> dilation = in TEST_F() local
267 exec_aten::ArrayRef<int64_t> dilation = in TEST_F() local
/external/pytorch/aten/src/ATen/native/miopen/
DConv_miopen.cpp35 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 …]
/external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/
DKernelUtils.cpp35 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()
/external/pytorch/aten/src/ATen/native/quantized/cpu/
Dqconv_prepack.cpp35 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 …]
/external/pytorch/aten/src/ATen/native/
DDilatedMaxPool3d.cpp28 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()
/external/pytorch/aten/src/ATen/native/vulkan/ops/
DConvolution.cpp60 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 …]
/external/pytorch/aten/src/ATen/native/mkldnn/
DConv.cpp26 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 …]
DConvPrepack.cpp25 std::vector<int64_t> dilation, in createConvPrePackOpContext()
49 const IntArrayRef dilation, in create()
104 IntArrayRef dilation, in _mkldnn_convolution_out()
153 IntArrayRef dilation, in mkldnn_convolution_out()
187 IntArrayRef dilation = context.dilation_; in get_output_sizes() local
/external/pytorch/aten/src/ATen/native/xnnpack/
DConvolution.cpp32 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()
/external/ComputeLibrary/tests/validation/fixtures/
DConvolutionLayerFixture.h68const Size2D &dilation, const ActivationLayerInfo &act_info, unsigned int num_groups) in configure_conv_function()
79const 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()
DDepthwiseConvolutionLayerFixture.h293 …orShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilation, unsigned int… in setup()
307 … channel, size_t batch, Size2D kernel_size, size_t depth_multiplier, Size2D dilation, Size2D strid… in setup()
446 … channel, size_t batch, Size2D kernel_size, size_t depth_multiplier, Size2D dilation, Size2D strid… in setup()
644 …orShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilation, unsigned int… in setup()
658 …orShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilation, unsigned int… in setup()
/external/pytorch/aten/src/ATen/functorch/
DBatchRulesPooling.cpp19 IntArrayRef padding, IntArrayRef dilation, bool ceil_mode, int64_t n, Func pooling_fn) { in max_pool_with_indices_batch_rule_helper()
44 IntArrayRef padding, IntArrayRef dilation, bool ceil_mode) { in max_pool3d_with_indices_batch_rule()
52 IntArrayRef padding, IntArrayRef dilation, bool ceil_mode) { in max_pool2d_with_indices_batch_rule()
/external/ComputeLibrary/tests/validation/reference/
DDepthwiseConvolutionLayer.cpp55 … 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()

12345678910