| 12345678910111213141516171819202122232425262728 |
- #pragma once
- #include <c10/util/Exception.h>
- #include <c10/util/string_view.h>
- namespace at {
- enum class padding_mode {
- reflect,
- replicate,
- circular,
- constant,
- };
- static inline c10::string_view padding_mode_string(padding_mode m) {
- switch (m) {
- case padding_mode::reflect:
- return "reflect";
- case padding_mode::replicate:
- return "replicate";
- case padding_mode::circular:
- return "circular";
- case padding_mode::constant:
- return "constant";
- }
- TORCH_CHECK(false, "Invalid padding mode (", static_cast<int64_t>(m), ")");
- }
- } // namespace at
|
