1 | #pragma once |
---|---|
2 | #include <c10/util/Exception.h> |
3 | #include <c10/util/string_view.h> |
4 | |
5 | namespace at { |
6 | |
7 | enum class padding_mode { |
8 | reflect, |
9 | replicate, |
10 | circular, |
11 | constant, |
12 | }; |
13 | |
14 | static inline c10::string_view padding_mode_string(padding_mode m) { |
15 | switch (m) { |
16 | case padding_mode::reflect: |
17 | return "reflect"; |
18 | case padding_mode::replicate: |
19 | return "replicate"; |
20 | case padding_mode::circular: |
21 | return "circular"; |
22 | case padding_mode::constant: |
23 | return "constant"; |
24 | } |
25 | TORCH_CHECK(false, "Invalid padding mode (", static_cast<int64_t>(m), ")"); |
26 | } |
27 | |
28 | } // namespace at |
29 |