| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Pooling.cpp | 34 IntArrayRef kernel_size, in mkldnn_max_pool2d()
35 IntArrayRef stride, in mkldnn_max_pool2d()
36 IntArrayRef padding, in mkldnn_max_pool2d()
37 IntArrayRef dilation, in mkldnn_max_pool2d()
44 IntArrayRef kernel_size, in mkldnn_max_pool3d()
45 IntArrayRef stride, in mkldnn_max_pool3d()
46 IntArrayRef padding, in mkldnn_max_pool3d()
47 IntArrayRef dilation, in mkldnn_max_pool3d()
54 IntArrayRef kernel_size, in mkldnn_avg_pool2d()
55 IntArrayRef stride, in mkldnn_avg_pool2d()
[all …]
|
| H A D | Conv.cpp | 26 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, int64_t groups) { in mkldnn_convolution()
49 const IntArrayRef& padding, in check_shape_forward()
50 const IntArrayRef& stride, in check_shape_forward()
51 const IntArrayRef& dilation, in check_shape_forward()
64 const IntArrayRef& weight_sizes = weight.sizes(); in check_shape_forward()
171 IntArrayRef stride, in _mkldnn_convolution_out()
172 IntArrayRef dilation, in _mkldnn_convolution_out()
173 IntArrayRef padding, in _mkldnn_convolution_out()
217 IntArrayRef padding, in _mkldnn_convolution()
218 IntArrayRef stride, in _mkldnn_convolution()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | ConvPlaceholders.cpp | 31 IntArrayRef padding, in cudnn_convolution()
32 IntArrayRef stride, in cudnn_convolution()
33 IntArrayRef dilation, in cudnn_convolution()
44 IntArrayRef padding, in cudnn_convolution_out()
45 IntArrayRef stride, in cudnn_convolution_out()
46 IntArrayRef dilation, in cudnn_convolution_out()
56 IntArrayRef input_size, in cudnn_convolution_backward_input()
59 IntArrayRef padding, in cudnn_convolution_backward_input()
60 IntArrayRef stride, in cudnn_convolution_backward_input()
61 IntArrayRef dilation, in cudnn_convolution_backward_input()
[all …]
|
| H A D | ConvShared.h | 48 IntArrayRef padding,
49 IntArrayRef stride,
50 IntArrayRef dilation,
68 IntArrayRef padding,
69 IntArrayRef stride,
70 IntArrayRef dilation,
80 IntArrayRef padding,
81 IntArrayRef stride,
82 IntArrayRef dilation,
92 IntArrayRef padding,
[all …]
|
| H A D | ConvShared.cpp | 107 IntArrayRef padding, in setConvolutionParams()
108 IntArrayRef stride, in setConvolutionParams()
109 IntArrayRef dilation, in setConvolutionParams()
210 IntArrayRef padding, in cudnn_convolution_forward_out()
211 IntArrayRef stride, in cudnn_convolution_forward_out()
212 IntArrayRef dilation, in cudnn_convolution_forward_out()
243 IntArrayRef padding, in cudnn_convolution()
244 IntArrayRef stride, in cudnn_convolution()
245 IntArrayRef dilation, in cudnn_convolution()
279 IntArrayRef padding, in cudnn_convolution_out()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ConvUtils.h | 14 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
15 at::IntArrayRef, at::IntArrayRef, std::array<bool, 2>);
18 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
19 at::IntArrayRef, at::IntArrayRef, std::array<bool, 3>);
22 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
23 at::IntArrayRef, int64_t, bool, bool, bool, std::array<bool,2>);
26 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
27 at::IntArrayRef, int64_t, std::array<bool,3>);
30 const at::Tensor&, const at::Tensor&, const at::Tensor&, at::IntArrayRef, at::IntArrayRef,
31 at::IntArrayRef, at::IntArrayRef, int64_t, bool, bool, bool, std::array<bool,2>);
[all …]
|
| H A D | NaiveDilatedConvolution.cpp | 31 const IntArrayRef input_size, in hvol2col()
32 const IntArrayRef output_size, in hvol2col()
33 const IntArrayRef kernel_size, in hvol2col()
34 const IntArrayRef stride_size, in hvol2col()
35 const IntArrayRef pad_size, in hvol2col()
36 const IntArrayRef dilation_size, in hvol2col()
89 const IntArrayRef input_size, in col2hvol()
90 const IntArrayRef output_size, in col2hvol()
91 const IntArrayRef kernel_size, in col2hvol()
92 const IntArrayRef stride_size, in col2hvol()
[all …]
|
| H A D | DilatedMaxPool3d.cpp | 25 IntArrayRef kernel_size, in max_pool3d_with_indices_out_cpu_template()
26 IntArrayRef stride, in max_pool3d_with_indices_out_cpu_template()
27 IntArrayRef padding, in max_pool3d_with_indices_out_cpu_template()
28 IntArrayRef dilation, in max_pool3d_with_indices_out_cpu_template()
117 IntArrayRef kernel_size, in max_pool3d_with_indices_backward_out_cpu_template()
118 IntArrayRef stride, in max_pool3d_with_indices_backward_out_cpu_template()
119 IntArrayRef padding, in max_pool3d_with_indices_backward_out_cpu_template()
120 IntArrayRef dilation, in max_pool3d_with_indices_backward_out_cpu_template()
201 IntArrayRef kernel_size, in max_pool3d_with_indices_out_cpu()
202 IntArrayRef stride, in max_pool3d_with_indices_out_cpu()
[all …]
|
| H A D | Pooling.cpp | 36 IntArrayRef x) { in check1d()
43 Tensor adaptive_avg_pool1d(const Tensor & self, IntArrayRef output_size) { in adaptive_avg_pool1d()
54 std::tuple<Tensor,Tensor> adaptive_max_pool1d(const Tensor & self, IntArrayRef output_size) { in adaptive_max_pool1d()
80 IntArrayRef kernel_size, in max_pool1d_with_indices()
81 IntArrayRef stride, in max_pool1d_with_indices()
82 IntArrayRef padding, in max_pool1d_with_indices()
83 IntArrayRef dilation, in max_pool1d_with_indices()
116 IntArrayRef kernel_size, in avg_pool1d()
117 IntArrayRef stride, in avg_pool1d()
118 IntArrayRef padding, in avg_pool1d()
[all …]
|
| H A D | Col2Im.cpp | 77 IntArrayRef output_size, in col2im_out_cpu_template()
78 IntArrayRef kernel_size, in col2im_out_cpu_template()
79 IntArrayRef dilation, in col2im_out_cpu_template()
80 IntArrayRef padding, in col2im_out_cpu_template()
81 IntArrayRef stride) { in col2im_out_cpu_template()
192 IntArrayRef output_size, in col2im_out_cpu()
193 IntArrayRef kernel_size, in col2im_out_cpu()
194 IntArrayRef dilation, in col2im_out_cpu()
195 IntArrayRef padding, in col2im_out_cpu()
196 IntArrayRef stride, in col2im_out_cpu()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| H A D | oneDNN.h | 24 IntArrayRef padding_front_top_left,
25 IntArrayRef padding_back_bottom_right,
26 IntArrayRef stride,
27 IntArrayRef dilation,
37 IntArrayRef diff_weight_aten_size,
38 IntArrayRef padding_front_top_left,
39 IntArrayRef padding_back_bottom_right,
40 IntArrayRef stride,
41 IntArrayRef dilation,
49 IntArrayRef padding_front_top_left,
[all …]
|
| /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()
41 IntArrayRef input_size, const at::Tensor& grad_output, const at::Tensor& weight, in miopen_convolution_backward_input()
42 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, int64_t groups, in miopen_convolution_backward_input()
48 IntArrayRef weight_size, const at::Tensor& grad_output, const at::Tensor& input, in miopen_convolution_backward_weight()
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()
81 IntArrayRef weight_size, const at::Tensor& grad_output, const at::Tensor& input, in miopen_convolution_transpose_backward_weight()
82 IntArrayRef padding, IntArrayRef stride, IntArrayRef dilation, int64_t groups, in miopen_convolution_transpose_backward_weight()
[all …]
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | kernel_ops_util.h | 23 inline int64_t val_at(IntArrayRef array, size_t i, int64_t default_value = 1) {
36 bool int_array_all_ge(IntArrayRef array, int64_t val);
38 bool stride_is_valid(IntArrayRef stride, size_t kernel_ndim, bool allow_empty);
41 IntArrayRef padding,
42 IntArrayRef kernel_size,
46 bool dilation_is_valid(IntArrayRef dilation, size_t kernel_ndim);
70 IntArrayRef kernel_sizes,
71 IntArrayRef stride,
72 IntArrayRef padding,
73 IntArrayRef dilation,
[all …]
|
| H A D | kernel_ops_util.cpp | 23 IntArrayRef array, in param_array_is_valid()
49 bool int_array_all_ge(IntArrayRef array, int64_t val) { in int_array_all_ge()
64 bool kernel_size_is_valid(IntArrayRef kernel_size, size_t kernel_ndim) { in kernel_size_is_valid()
73 bool stride_is_valid(IntArrayRef stride, size_t kernel_ndim, bool allow_empty) { in stride_is_valid()
79 IntArrayRef padding, in padding_is_valid()
80 IntArrayRef kernel_size, in padding_is_valid()
108 bool dilation_is_valid(IntArrayRef dilation, size_t kernel_ndim) { in dilation_is_valid()
114 IntArrayRef output_padding, in output_padding_is_valid()
115 IntArrayRef stride, in output_padding_is_valid()
116 IntArrayRef dilation, in output_padding_is_valid()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | EmptyTensor.h | 29 IntArrayRef sizes,
37 IntArrayRef sizes,
38 IntArrayRef strides,
48 IntArrayRef size,
62 IntArrayRef size,
63 IntArrayRef stride,
76 IntArrayRef size,
82 IntArrayRef size,
89 TORCH_API TensorBase empty_cpu(IntArrayRef size, const TensorOptions& options);
92 IntArrayRef size,
[all …]
|
| H A D | VmapModeRegistrations.cpp | 64 …m.impl("normal.float_float", unsupportedRandomOp<double, double, IntArrayRef, std::optional<Genera… in TORCH_LIBRARY_IMPL()
65 …m.impl("normal.float_float_out", unsupportedRandomOp_<double, double, IntArrayRef, std::optional<G… in TORCH_LIBRARY_IMPL()
80 m.impl("rand", unsupportedRandomOp<IntArrayRef, TENSOROPTIONS>); in TORCH_LIBRARY_IMPL()
81 …m.impl("rand.generator", unsupportedRandomOp<IntArrayRef, std::optional<Generator>, TENSOROPTIONS>… in TORCH_LIBRARY_IMPL()
82 m.impl("rand.names", unsupportedRandomOp<IntArrayRef, std::optional<DimnameList>, TENSOROPTIONS>); in TORCH_LIBRARY_IMPL()
83 …m.impl("rand.generator_with_names", unsupportedRandomOp<IntArrayRef, std::optional<Generator>, std… in TORCH_LIBRARY_IMPL()
84 m.impl("rand.out", unsupportedRandomOp_<IntArrayRef, Tensor&>); in TORCH_LIBRARY_IMPL()
85 …m.impl("rand.generator_out", unsupportedRandomOp_<IntArrayRef, std::optional<Generator>, Tensor&>); in TORCH_LIBRARY_IMPL()
87 m.impl("randn", unsupportedRandomOp<IntArrayRef, TENSOROPTIONS>); in TORCH_LIBRARY_IMPL()
88 …m.impl("randn.generator", unsupportedRandomOp<IntArrayRef, std::optional<Generator>, TENSOROPTIONS… in TORCH_LIBRARY_IMPL()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | NaiveDilatedConvolution.cu | 35 const IntArrayRef input_size, in hvol2col()
36 const IntArrayRef output_size, in hvol2col()
37 const IntArrayRef kernel_size, in hvol2col()
38 const IntArrayRef stride_size, in hvol2col()
39 const IntArrayRef pad_size, in hvol2col()
40 const IntArrayRef dilation_size, in hvol2col()
94 const IntArrayRef input_size, in col2hvol()
95 const IntArrayRef output_size, in col2hvol()
96 const IntArrayRef kernel_size, in col2hvol()
97 const IntArrayRef stride_size, in col2hvol()
[all …]
|
| H A D | NaiveConvolutionTranspose3d.cu | 177 IntArrayRef kernel_size, in slow_conv_transpose3d_out_cuda_template()
179 IntArrayRef stride, in slow_conv_transpose3d_out_cuda_template()
180 IntArrayRef padding, in slow_conv_transpose3d_out_cuda_template()
181 IntArrayRef output_padding, in slow_conv_transpose3d_out_cuda_template()
182 IntArrayRef dilation) { in slow_conv_transpose3d_out_cuda_template()
399 IntArrayRef kernel_size, in slow_conv_transpose3d_backward_out_cuda_template()
400 IntArrayRef stride, in slow_conv_transpose3d_backward_out_cuda_template()
401 IntArrayRef padding, in slow_conv_transpose3d_backward_out_cuda_template()
402 IntArrayRef output_padding, in slow_conv_transpose3d_backward_out_cuda_template()
403 IntArrayRef dilation) { in slow_conv_transpose3d_backward_out_cuda_template()
[all …]
|
| H A D | NaiveConvolutionTranspose2d.cu | 152 IntArrayRef kernel_size, in slow_conv_transpose2d_out_cuda_template()
154 IntArrayRef stride, in slow_conv_transpose2d_out_cuda_template()
155 IntArrayRef padding, in slow_conv_transpose2d_out_cuda_template()
156 IntArrayRef output_padding, in slow_conv_transpose2d_out_cuda_template()
157 IntArrayRef dilation) { in slow_conv_transpose2d_out_cuda_template()
308 IntArrayRef kernel_size, in slow_conv_transpose2d_backward_out_cuda_template()
309 IntArrayRef stride, in slow_conv_transpose2d_backward_out_cuda_template()
310 IntArrayRef padding, in slow_conv_transpose2d_backward_out_cuda_template()
311 IntArrayRef output_padding, in slow_conv_transpose2d_backward_out_cuda_template()
312 IntArrayRef dilation) { in slow_conv_transpose2d_backward_out_cuda_template()
[all …]
|
| H A D | Col2Im.cu | 29 IntArrayRef output_size, in col2im_out_cuda_template()
30 IntArrayRef kernel_size, in col2im_out_cuda_template()
31 IntArrayRef dilation, in col2im_out_cuda_template()
32 IntArrayRef padding, in col2im_out_cuda_template()
33 IntArrayRef stride) { in col2im_out_cuda_template()
146 IntArrayRef output_size, in col2im_out_cuda()
147 IntArrayRef kernel_size, in col2im_out_cuda()
148 IntArrayRef dilation, in col2im_out_cuda()
149 IntArrayRef padding, in col2im_out_cuda()
150 IntArrayRef stride, in col2im_out_cuda()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | AdaptivePooling.mm | 63 Tensor& adaptive_avg_pool2d_out_mps(const Tensor& input, IntArrayRef output_size, Tensor& output) {
86 IntArrayRef({kernel_sizeH, kernel_sizeW}),
87 IntArrayRef({strideH, strideW}),
88 IntArrayRef({0, 0}),
98 phony_grad.resize_(IntArrayRef(phony_shape));
101 IntArrayRef({kernel_sizeH, kernel_sizeW}),
102 IntArrayRef({strideH, strideW}),
103 IntArrayRef({0, 0}),
114 Tensor adaptive_avg_pool2d_mps(at::Tensor const& input, IntArrayRef output_size) {
115 IntArrayRef output_shape;
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | Shim.cpp | 35 const IntArrayRef, in use_convolution2d() argument
36 const IntArrayRef, in use_convolution2d()
37 const IntArrayRef, in use_convolution2d()
47 const IntArrayRef, in convolution2d() argument
48 const IntArrayRef, in convolution2d()
49 const IntArrayRef, in convolution2d()
70 const IntArrayRef, in use_max_pool2d() argument
71 const IntArrayRef, in use_max_pool2d()
72 IntArrayRef, in use_max_pool2d()
73 const IntArrayRef, in use_max_pool2d()
[all …]
|
| H A D | Engine.h | 16 const IntArrayRef padding,
17 const IntArrayRef stride,
18 const IntArrayRef dilation,
26 const IntArrayRef padding,
27 const IntArrayRef stride,
28 const IntArrayRef dilation,
51 const IntArrayRef kernel,
52 const IntArrayRef padding,
53 IntArrayRef stride,
54 const IntArrayRef dilation,
[all …]
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_constant_pad_nd_test.cpp | 19 using exec_aten::IntArrayRef;
30 const IntArrayRef padding, in op_constant_pad_nd_out()
77 IntArrayRef padding_ref = IntArrayRef(padding.data(), padding.size()); in test_constant_pad_nd_out_dim2()
129 IntArrayRef padding_ref = IntArrayRef(padding.data(), padding.size()); in test_constant_pad_nd_out_dim1()
182 IntArrayRef padding_ref = IntArrayRef(padding.data(), padding.size()); in test_constant_pad_nd_out_dim0()
234 IntArrayRef padding_ref = IntArrayRef(padding.data(), padding.size()); in test_constant_pad_nd_out_dim12()
287 IntArrayRef padding_ref = IntArrayRef(padding.data(), padding.size()); in test_constant_pad_nd_out_dim02()
343 IntArrayRef padding_ref = IntArrayRef(padding.data(), padding.size()); in test_constant_pad_nd_out_dim012()
408 IntArrayRef padding_ref = IntArrayRef(padding.data(), padding.size()); in TEST_F()
424 IntArrayRef padding_ref = IntArrayRef(padding.data(), padding.size()); in TEST_F()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Convolution.cpp | 35 inline bool is_depthwise(const IntArrayRef weight_size, const int64_t groups) { in is_depthwise()
46 inline bool is_pointwise(const IntArrayRef weight_size) { in is_pointwise()
57 const IntArrayRef weight_size, in determine_method()
58 const IntArrayRef stride, in determine_method()
59 const IntArrayRef padding, in determine_method()
60 const IntArrayRef dilation, in determine_method()
286 const IntArrayRef kernel_size, in get_shader()
287 const IntArrayRef stride, in get_shader()
288 const IntArrayRef padding, in get_shader()
289 const IntArrayRef dilation, in get_shader()
[all …]
|