Lines Matching refs:rootOp

202   if (isEmpty() || tiledRootAndFusedOpsLoops.count(rootOp) == 0)  in isValid()
206   if (tileLoopOps.size() != tiledRootAndFusedOpsLoops[rootOp].size())  in isValid()
210   if (rootOp->getParentOp() != tileLoopOps.back())  in isValid()
296       tileLinalgOp(rewriter, rootOp, tilingOptions);  in tileRootOp()
306     rootOp->replaceAllUsesWith(tiledRootOp->tensorResults);  in tileRootOp()
309   if (tiledRootAndFusedOpsLoops.count(rootOp) != 0) {  in tileRootOp()
311         tiledRootAndFusedOpsLoops[rootOp];  in tileRootOp()
315   rootOp = tiledRootOp->op;  in tileRootOp()
321     tiledRootAndFusedOpsLoops[rootOp].push_back(tileInterchange[en.index()]);  in tileRootOp()
349   if (sliceOp->getBlock() != rootOp->getBlock() ||  in fuseProducer()
350       consumerOp->getBlock() != rootOp->getBlock())  in fuseProducer()