DimVector.h 425 B

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