size_ Class — pytorch Architecture

Source Code

aten/src/ATen/core/Array.h lines 11–46

template <typename T, int size_>
struct Array {
  // NOLINTNEXTLINE(*c-array*)
  T data[size_];

  C10_HOST_DEVICE T operator[](int i) const {
    return data[i];
  }
  C10_HOST_DEVICE T& operator[](int i) {
    return data[i];
  }
#if defined(USE_ROCM)
  C10_HOST_DEVICE Array() = default;
  C10_HOST_DEVICE Array(const Array&) = default;
  C10_HOST_DEVICE Array& operator=(const Array&) = default;
  C10_HOST_DEVICE Array(Array&&) = default;
  C10_HOST_DEVICE Array& operator=(Array&&) = default;
  C10_HOST_DEVICE ~Array() = default;
#else
  Array() = default;
  Array(const Array&) = default;
  Array& operator=(const Array&) = default;
  Array(Array&&) noexcept = default;
  Array& operator=(Array&&) noexcept = default;
  ~Array() = default;
#endif
  static constexpr int size() {
    return size_;
  }
  // Fill the array with x.
  C10_HOST_DEVICE Array(T x) {
    for (int i = 0; i < size_; i++) {
      data[i] = x;
    }
  }
};