xref: /aosp_15_r20/external/pytorch/aten/src/ATen/PadNd.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
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 
padding_mode_string(padding_mode m)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