You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
593 B
29 lines
593 B
5 months ago
|
#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
|