39 static void getLoadAndStoreMemRefAccesses(Operation *opA,  in getLoadAndStoreMemRefAccesses()
41   opA->walk([&](Operation *op) {  in getLoadAndStoreMemRefAccesses()
54 static bool isDependentLoadOrStoreOp(Operation *op,  in isDependentLoadOrStoreOp()
67 static Operation *getFirstDependentOpInRange(Operation *opA, Operation *opB) {  in getFirstDependentOpInRange()
76   Operation *firstDepOp = nullptr;  in getFirstDependentOpInRange()
79     Operation *opX = &(*it);  in getFirstDependentOpInRange()
80     opX->walk([&](Operation *op) {  in getFirstDependentOpInRange()
93 static Operation *getLastDependentOpInRange(Operation *opA, Operation *opB) {  in getLastDependentOpInRange()
104   Operation *lastDepOp = nullptr;  in getLastDependentOpInRange()
107     Operation *opX = &(*it);  in getLastDependentOpInRange()
108     opX->walk([&](Operation *op) {  in getLastDependentOpInRange()
117         for (Operation *user : value.getUsers()) {  in getLastDependentOpInRange()
138 static Operation *getFusedLoopNestInsertionPoint(AffineForOp srcForOp,  in getFusedLoopNestInsertionPoint()
180                      SmallVectorImpl<Operation *> &loadAndStoreOps) {  in gatherLoadsAndStores()
182   forOp.walk([&](Operation *op) {  in gatherLoadsAndStores()
195 static unsigned getMaxLoopDepth(ArrayRef<Operation *> srcOps,  in getMaxLoopDepth()
196                                 ArrayRef<Operation *> dstOps) {  in getMaxLoopDepth()
206   SmallVector<Operation *, 4> targetDstOps;  in getMaxLoopDepth()
207   for (Operation *dstOp : dstOps) {  in getMaxLoopDepth()
223              [&](Operation *op) { return isa<AffineReadOpInterface>(op); }))  in getMaxLoopDepth()
291   SmallVector<Operation *, 4> opsA;  in canFuseLoops()
298   SmallVector<Operation *, 4> opsB;  in canFuseLoops()
323   SmallVector<Operation *, 4> strategyOpsA;  in canFuseLoops()
333     for (Operation *op : opsA) {  in canFuseLoops()
341     for (Operation *op : opsA) {  in canFuseLoops()
390   SetVector<Operation *> forwardSlice;  in promoteSingleIterReductionLoop()
393       SetVector<Operation *> tmpForwardSlice;  in promoteSingleIterReductionLoop()
407     for (Operation *op : llvm::reverse(forwardSlice))  in promoteSingleIterReductionLoop()
460   llvm::SmallDenseMap<Operation *, uint64_t, 8> sliceTripCountMap;  in fuseLoops()
534     Operation *forOp, LoopNestStats &stats,  in getComputeCostHelper()
535     llvm::SmallDenseMap<Operation *, uint64_t, 8> *tripCountOverrideMap,  in getComputeCostHelper()  argument
536     DenseMap<Operation *, int64_t> *computeCostMap) {  in getComputeCostHelper()  argument
584   llvm::SmallDenseMap<Operation *, uint64_t, 8> sliceTripCountMap;  in getFusionComputeCost()
585   DenseMap<Operation *, int64_t> computeCostMap;  in getFusionComputeCost()
603     srcForOp.walk([&](Operation *op) {  in getFusionComputeCost()
651     ArrayRef<Operation *> srcOps, ArrayRef<Operation *> dstOps,  in gatherProducerConsumerMemrefs()
655   for (Operation *op : srcOps)  in gatherProducerConsumerMemrefs()
661   for (Operation *op : dstOps)  in gatherProducerConsumerMemrefs()