/usr/local/lib64/python3.6/site-packages/torch/include/ATen/core
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/core/DimVector.h (247B)
#pragma once
#include
#include
namespace at {
constexpr size_t kDimVectorStaticSize = 5;
/// A container for sizes or strides
using DimVector = SmallVector;
} // namespace at