Lines Matching defs:indexOp
241 for (IndexOp indexOp :
245 consumerToProducerLoopsMap.getSubMap(indexOp.getDim()), fusedIndices);
246 mapper.map(indexOp.getResult(), newIndex);
759 for (IndexOp indexOp :
762 expansionInfo.getExpandedDims(indexOp.getDim());
767 expandedDims.front() == (int64_t)indexOp.getDim())
772 rewriter.setInsertionPointAfter(indexOp);
774 expansionInfo.getExpandedShapeOfDim(indexOp.getDim()).drop_front();
786 indexOp.getLoc(), idx + acc * std::get<0>(it),
789 rewriter.replaceOp(indexOp, newIndex);
1583 for (auto indexOp : indexOps) {
1584 auto dim = indexOp.getDim();
1585 rewriter.replaceOp(indexOp, indexReplacementVals[dim]);