xref: /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/padding.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #include <torch/nn/options/padding.h>
2 
3 namespace torch {
4 namespace nn {
5 
6 template struct ReflectionPadOptions<1>;
7 template struct ReflectionPadOptions<2>;
8 
9 template struct ReplicationPadOptions<1>;
10 template struct ReplicationPadOptions<2>;
11 template struct ReplicationPadOptions<3>;
12 
13 template struct ConstantPadOptions<1>;
14 template struct ConstantPadOptions<2>;
15 template struct ConstantPadOptions<3>;
16 
17 namespace functional {
18 
PadFuncOptions(std::vector<int64_t> pad)19 PadFuncOptions::PadFuncOptions(std::vector<int64_t> pad)
20     : pad_(std::move(pad)) {}
21 
22 } // namespace functional
23 
24 } // namespace nn
25 } // namespace torch
26