c10/util/DimVector.h

#pragma once

#include <c10/util/SmallVector.h>
#include <cstdint>

namespace c10 {

constexpr size_t kDimVectorStaticSize = 5;

/// A container for sizes or strides
using DimVector = SmallVector<int64_t, kDimVectorStaticSize>;

} // namespace c10