Home
last modified time | relevance | path

Searched refs:ExpandingArray (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h16 ExpandingArray<1> kernel_size, in avg_pool1d()
17 ExpandingArray<1> stride, in avg_pool1d()
18 ExpandingArray<1> padding, in avg_pool1d()
55 ExpandingArray<2> kernel_size, in avg_pool2d()
56 ExpandingArray<2> stride, in avg_pool2d()
57 ExpandingArray<2> padding, in avg_pool2d()
102 ExpandingArray<3> kernel_size, in avg_pool3d()
103 ExpandingArray<3> stride, in avg_pool3d()
104 ExpandingArray<3> padding, in avg_pool3d()
151 ExpandingArray<1> kernel_size, in max_pool1d()
[all …]
H A Dfold.h13 ExpandingArray<2> output_size, in fold()
14 ExpandingArray<2> kernel_size, in fold()
15 ExpandingArray<2> dilation, in fold()
16 ExpandingArray<2> padding, in fold()
17 ExpandingArray<2> stride) { in fold()
61 ExpandingArray<2> kernel_size, in unfold()
62 ExpandingArray<2> dilation, in unfold()
63 ExpandingArray<2> padding, in unfold()
64 ExpandingArray<2> stride) { in unfold()
H A Dconv.h22 IntArrayRef padding_unwrap(const ExpandingArray<D>& array) { in padding_unwrap()
30 ExpandingArray<1> stride, in conv1d()
32 ExpandingArray<1> dilation, in conv1d()
76 ExpandingArray<2> stride, in conv2d()
78 ExpandingArray<2> dilation, in conv2d()
122 ExpandingArray<3> stride, in conv3d()
124 ExpandingArray<3> dilation, in conv3d()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/
H A Dpooling.h14 AvgPoolOptions(ExpandingArray<D> kernel_size) in AvgPoolOptions()
18 TORCH_ARG(ExpandingArray<D>, kernel_size);
21 TORCH_ARG(ExpandingArray<D>, stride);
24 TORCH_ARG(ExpandingArray<D>, padding) = 0;
109 MaxPoolOptions(ExpandingArray<D> kernel_size) in MaxPoolOptions()
113 TORCH_ARG(ExpandingArray<D>, kernel_size);
116 TORCH_ARG(ExpandingArray<D>, stride);
119 TORCH_ARG(ExpandingArray<D>, padding) = 0;
122 TORCH_ARG(ExpandingArray<D>, dilation) = 1;
206 using AdaptiveMaxPool1dOptions = AdaptiveMaxPoolOptions<ExpandingArray<1>>;
[all …]
H A Dconv.h23 std::variant<ExpandingArray<D>, enumtype::kValid, enumtype::kSame>;
32 ExpandingArray<D> kernel_size) in ConvNdOptions()
49 TORCH_ARG(ExpandingArray<D>, kernel_size);
55 TORCH_ARG(ExpandingArray<D>, stride) = 1;
72 TORCH_ARG(ExpandingArray<D>, dilation) = 1;
83 TORCH_ARG(ExpandingArray<D>, output_padding) = 0;
111 ExpandingArray<D> kernel_size) in ConvOptions()
128 TORCH_ARG(ExpandingArray<D>, kernel_size);
134 TORCH_ARG(ExpandingArray<D>, stride) = 1;
151 TORCH_ARG(ExpandingArray<D>, dilation) = 1;
[all …]
H A Dfold.h19 FoldOptions(ExpandingArray<2> output_size, ExpandingArray<2> kernel_size) in FoldOptions()
26 TORCH_ARG(ExpandingArray<2>, output_size);
29 TORCH_ARG(ExpandingArray<2>, kernel_size);
33 TORCH_ARG(ExpandingArray<2>, dilation) = 1;
37 TORCH_ARG(ExpandingArray<2>, padding) = 0;
40 TORCH_ARG(ExpandingArray<2>, stride) = 1;
66 UnfoldOptions(ExpandingArray<2> kernel_size) in UnfoldOptions()
70 TORCH_ARG(ExpandingArray<2>, kernel_size);
74 TORCH_ARG(ExpandingArray<2>, dilation) = 1;
78 TORCH_ARG(ExpandingArray<2>, padding) = 0;
[all …]
H A Dpadding.h15 ReflectionPadOptions(ExpandingArray<D * 2> padding) : padding_(padding) {} in ReflectionPadOptions()
25 TORCH_ARG(ExpandingArray<D * 2>, padding);
57 ReplicationPadOptions(ExpandingArray<D * 2> padding) : padding_(padding) {} in ReplicationPadOptions()
68 TORCH_ARG(ExpandingArray<D * 2>, padding);
99 ZeroPadOptions(ExpandingArray<D * 2> padding) : padding_(padding) {} in ZeroPadOptions()
110 TORCH_ARG(ExpandingArray<D * 2>, padding);
142 ConstantPadOptions(ExpandingArray<D * 2> padding, double value) in ConstantPadOptions()
154 TORCH_ARG(ExpandingArray<D * 2>, padding);
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Dconv.h70 [&](const ExpandingArray<D>& pad) { in reset()
126 [&](const ExpandingArray<D>& pad) { in pretty_print()
127 if (*pad != *ExpandingArray<D>(0)) { in pretty_print()
132 if (*options.dilation() != *ExpandingArray<D>(1)) { in pretty_print()
135 if (*options.output_padding() != *ExpandingArray<D>(0)) { in pretty_print()
186 ExpandingArray<1> kernel_size) in Conv1dImpl()
218 ExpandingArray<2> kernel_size) in Conv2dImpl()
253 ExpandingArray<3> kernel_size) in Conv3dImpl()
277 std::holds_alternative<ExpandingArray<D>>(this->options.padding()), in ConvTransposeNdImpl()
289 if (*pad != *ExpandingArray<D>(0)) { in pretty_print()
[all …]
H A Dpooling.h18 AvgPoolImpl(ExpandingArray<D> kernel_size) in AvgPoolImpl()
115 MaxPoolImpl(ExpandingArray<D> kernel_size) in MaxPoolImpl()
258 : public AdaptiveMaxPoolImpl<1, ExpandingArray<1>, AdaptiveMaxPool1dImpl> {
260 using AdaptiveMaxPoolImpl<1, ExpandingArray<1>, AdaptiveMaxPool1dImpl>::
391 : public AdaptiveAvgPoolImpl<1, ExpandingArray<1>, AdaptiveAvgPool1dImpl> {
393 using AdaptiveAvgPoolImpl<1, ExpandingArray<1>, AdaptiveAvgPool1dImpl>::
478 MaxUnpoolImpl(ExpandingArray<D> kernel_size) in MaxUnpoolImpl()
604 FractionalMaxPool2dImpl(ExpandingArray<2> kernel_size) in FractionalMaxPool2dImpl()
649 FractionalMaxPool3dImpl(ExpandingArray<3> kernel_size) in FractionalMaxPool3dImpl()
683 LPPoolImpl(double norm_type, ExpandingArray<D> kernel_size) in LPPoolImpl()
H A Dpadding.h16 ReflectionPadImpl(ExpandingArray<D * 2> padding) in ReflectionPadImpl()
119 ReplicationPadImpl(ExpandingArray<D * 2> padding) in ReplicationPadImpl()
221 ZeroPadImpl(ExpandingArray<D * 2> padding) in ZeroPadImpl()
284 ConstantPadImpl(ExpandingArray<D * 2> padding, double value) in ConstantPadImpl()
H A Dfold.h27 FoldImpl(ExpandingArray<2> output_size, ExpandingArray<2> kernel_size) in FoldImpl()
64 UnfoldImpl(ExpandingArray<2> kernel_size) in UnfoldImpl()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/
H A Dexpanding_array.h24 class ExpandingArray {
29 /*implicit*/ ExpandingArray(std::initializer_list<T> list) in ExpandingArray() function
30 : ExpandingArray(at::ArrayRef<T>(list)) {} in ExpandingArray()
35 /*implicit*/ ExpandingArray(std::vector<T> vec) in ExpandingArray() function
36 : ExpandingArray(at::ArrayRef<T>(vec)) {} in ExpandingArray()
41 /*implicit*/ ExpandingArray(at::ArrayRef<T> values) { in ExpandingArray() function
52 /*implicit*/ ExpandingArray(T single_size) { in ExpandingArray() function
57 /*implicit*/ ExpandingArray(const std::array<T, D>& values) in ExpandingArray() function
99 const ExpandingArray<D, T>& expanding_array) {
113 : public ExpandingArray<D, std::optional<T>> {
[all …]
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dexpanding-array.cpp15 torch::ExpandingArray<5> e({1, 2, 3, 4, 5}); in TEST_F()
23 torch::ExpandingArray<5> e(std::vector<int64_t>{1, 2, 3, 4, 5}); in TEST_F()
31 torch::ExpandingArray<5> e(std::array<int64_t, 5>({1, 2, 3, 4, 5})); in TEST_F()
39 torch::ExpandingArray<5> e(5); in TEST_F()
50 torch::ExpandingArray<5>({1, 2, 3, 4, 5, 6, 7}), in TEST_F()
58 torch::ExpandingArray<5>(std::vector<int64_t>({1, 2, 3, 4, 5, 6, 7})), in TEST_F()
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dpooling.cpp14 template struct AdaptiveMaxPoolOptions<ExpandingArray<1>>;
18 template struct AdaptiveAvgPoolOptions<ExpandingArray<1>>;
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Dconv.cpp173 const ExpandingArray<D>& stride, in _output_padding()
174 const ExpandingArray<D>& padding, in _output_padding()
175 const ExpandingArray<D>& kernel_size) { in _output_padding()
H A Dpooling.cpp178 template class AdaptiveMaxPoolImpl<1, ExpandingArray<1>, AdaptiveMaxPool1dImpl>;
202 template class AdaptiveAvgPoolImpl<1, ExpandingArray<1>, AdaptiveAvgPool1dImpl>;