Searched refs:srcStrides (Results 1 – 3 of 3) sorted by relevance
71 int64_t *srcStrides = static_cast<int64_t *>(alloca(sizeof(int64_t) * rank)); in memrefCopy() local77 srcStrides[rankp] = src.strides[rankp] * elemSize; in memrefCopy()89 readIndex += srcStrides[axis]; in memrefCopy()100 readIndex -= src.sizes[axis] * srcStrides[axis]; in memrefCopy()
1709 SmallVector<int64_t> srcStrides; in computeExpandedLayoutMap() local1710 if (failed(getStridesAndOffset(srcType, srcStrides, srcOffset))) in computeExpandedLayoutMap()1712 assert(srcStrides.size() == reassociation.size() && "invalid reassociation"); in computeExpandedLayoutMap()1729 for (auto it : llvm::reverse(llvm::zip(reassociation, srcStrides))) { in computeExpandedLayoutMap()1826 SmallVector<int64_t> srcStrides; in computeCollapsedLayoutMap() local1828 if (failed(getStridesAndOffset(srcType, srcStrides, srcOffset))) in computeCollapsedLayoutMap()1843 resultStrides.push_back(srcStrides[ref.back()]); in computeCollapsedLayoutMap()1876 auto srcStride = Wrapper::stride(srcStrides[idx - 1]); in computeCollapsedLayoutMap()
2620 SmallVector<int64_t> srcStrides; in matchAndRewrite() local2622 if (failed(getStridesAndOffset(srcType, srcStrides, srcOffset))) in matchAndRewrite()2626 for (size_t i = 1; i < srcStrides.size(); ++i) { in matchAndRewrite()2628 if (srcStrides[dim] == 1) { in matchAndRewrite()