xref: /aosp_15_r20/external/pytorch/aten/src/ATen/PadNd.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

#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