DimVector.h 444 B

1234567891011121314151617
  1. #pragma once
  2. #include <c10/core/SymInt.h>
  3. #include <c10/core/impl/SizesAndStrides.h>
  4. #include <c10/util/SmallVector.h>
  5. #include <cstddef>
  6. #include <cstdint>
  7. namespace c10 {
  8. constexpr size_t kDimVectorStaticSize = C10_SIZES_AND_STRIDES_MAX_INLINE_SIZE;
  9. /// A container for sizes or strides
  10. using DimVector = SmallVector<int64_t, kDimVectorStaticSize>;
  11. using SymDimVector = SmallVector<c10::SymInt, kDimVectorStaticSize>;
  12. } // namespace c10