Viewing File: /home/ubuntu/combine_ai/combine/lib/python3.10/site-packages/torch/include/ATen/PadNd.h

#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
Back to Directory File Manager