Searched refs:expandedShape (Results 1 – 2 of 2) sorted by relevance
441 ArrayRef<int64_t> expandedShape = in matchAndRewrite() local449 while (expandedDim < expandedShape.size() && in matchAndRewrite()452 int64_t srcSize = expandedShape[expandedDim]; in matchAndRewrite()453 while (srcSize < dstSize && expandedDim < expandedShape.size()) { in matchAndRewrite()456 srcSize *= expandedShape[expandedDim]; in matchAndRewrite()465 while (expandedDim < expandedShape.size() && in matchAndRewrite()466 expandedShape[expandedDim] == 1) { in matchAndRewrite()476 if (expandedDim != expandedShape.size()) in matchAndRewrite()482 return AffineMap::get(expandedShape.size(), 0, exprs, in matchAndRewrite()
453 ArrayRef<int64_t> expandedShape = expandedType.getShape(); in fuseWithReshapeByExpansion() local469 expandedShape.slice(foldedDims.getDimPosition(0), numFoldedDims[pos]); in fuseWithReshapeByExpansion()484 for (auto &expandedShape : expandedDimsShape) { in fuseWithReshapeByExpansion() local485 if (expandedShape.size() == 1) in fuseWithReshapeByExpansion()488 std::next(expandedShape.begin()), expandedShape.end())) { in fuseWithReshapeByExpansion()