Searched refs:TensorAccessBoundsInfo (Results 1 – 4 of 4) sorted by relevance
19 struct TORCH_API TensorAccessBoundsInfo { struct26 std::unordered_map<BufPtr, std::vector<TensorAccessBoundsInfo>>; argument45 const std::vector<TensorAccessBoundsInfo>& infos);
35 std::vector<TensorAccessBoundsInfo>& infos = ret[buf]; in mergeTensorAccesses()61 TensorAccessBoundsInfo info; in mergeTensorAccesses()164 const std::vector<TensorAccessBoundsInfo>& infos) { in getBoundExtents()170 for (const TensorAccessBoundsInfo& p : infos) { in getBoundExtents()202 const std::vector<TensorAccessBoundsInfo>& bounds, in convertBounds()277 static IndexBounds getIndexBounds(const TensorAccessBoundsInfo& tabi) { in getIndexBounds()291 const std::vector<TensorAccessBoundsInfo>& vTABI, in getIndexBounds()
2858 TensorAccessBoundsInfo& info = bounds_it->second[0]; in cacheAccesses()3147 for (const TensorAccessBoundsInfo& p : bounds_it->second) { in computeAt()
25 const TensorAccessBoundsInfo& access_info, in verifyConstBounds()519 TensorAccessBoundsInfo& after = pair.second[i]; in TEST()520 TensorAccessBoundsInfo& before = beforeIt->second[i]; in TEST()535 TensorAccessBoundsInfo& first = pair.second[0]; in TEST()536 TensorAccessBoundsInfo& second = pair.second[1]; in TEST()