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