1#pragma once
2#include <c10/util/Exception.h>
3#include <c10/util/string_view.h>
4
5namespace at {
6
7enum class padding_mode {
8 reflect,
9 replicate,
10 circular,
11 constant,
12};
13
14static 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