29 lines
593 B
C
29 lines
593 B
C
|
#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
|