Searched refs:srcShape (Results 1 – 7 of 7) sorted by relevance
115 auto srcShape = multiReductionOp.getSourceVectorType().getShape(); in matchAndRewrite() local135 reductionShapes.push_back(srcShape[i]); in matchAndRewrite()138 parallelShapes.push_back(srcShape[i]); in matchAndRewrite()240 ArrayRef<int64_t> srcShape = in matchAndRewrite() local248 for (int64_t i = 0; i < srcShape[0]; i++) { in matchAndRewrite()315 auto srcShape = srcVectorType.getShape(); in matchAndRewrite() local316 auto castedType = VectorType::get(ArrayRef<int64_t>{1, srcShape.back()}, in matchAndRewrite()
338 ArrayRef<int64_t> srcShape, ArrayRef<int64_t> resultShape) const { in findCollapsingReassociation()342 return {getReassociationIndicesForCollapse(srcShape, resultShape)}; in findCollapsingReassociation()347 auto srcSubShape = srcShape.slice(srcIndices.front(), srcIndices.size()); in findCollapsingReassociation()
280 auto srcShape = srcType.getShape(); in matchAndRewrite() local281 int64_t m = srcShape[srcGtOneDims[0]], n = srcShape[srcGtOneDims[1]]; in matchAndRewrite()
913 PatternRewriter &rewriter, ArrayRef<int64_t> srcShape, in createReassociationMapsForCollapse() argument920 for (int i = 0, s = srcShape.size(); i < s; ++i) in createReassociationMapsForCollapse()933 while (currSrcDim < srcShape.size() && currDstDim < dstShape.size()) { in createReassociationMapsForCollapse()935 int64_t srcSize = srcShape[currSrcDim]; in createReassociationMapsForCollapse()936 while (srcSize < dstSize && currSrcDim < srcShape.size()) { in createReassociationMapsForCollapse()939 srcSize *= srcShape[currSrcDim]; in createReassociationMapsForCollapse()947 while (currSrcDim < srcShape.size() && srcShape[currSrcDim] == 1) { in createReassociationMapsForCollapse()958 return currSrcDim == srcShape.size() && currDstDim == dstShape.size(); in createReassociationMapsForCollapse()
1827 auto srcShape = srcType.getShape(); in computeCollapsedLayoutMap() local1840 while (srcShape[ref.back()] == 1 && ref.size() > 1) in computeCollapsedLayoutMap()1842 if (!ShapedType::isDynamic(srcShape[ref.back()]) || ref.size() == 1) { in computeCollapsedLayoutMap()1860 stride = stride * Wrapper::size(srcShape[idx]); in computeCollapsedLayoutMap()1868 if (srcShape[idx] == 1) in computeCollapsedLayoutMap()
1476 auto srcShape = srcType.getShape(); in fillInStridesForCollapsedMemDescriptor() local1481 while (srcShape[ref.back()] == 1 && ref.size() > 1) in fillInStridesForCollapsedMemDescriptor()1483 if (!ShapedType::isDynamic(srcShape[ref.back()]) || ref.size() == 1) { in fillInStridesForCollapsedMemDescriptor()1527 if (srcShape[srcIndex] == 1 && srcIndex != ref.front()) in fillInStridesForCollapsedMemDescriptor()
4855 ArrayRef<int64_t> srcShape = srcType.getShape(); in verify() local4860 expectedShape.push_back(srcShape[i]); in verify()