Home
last modified time | relevance | path

Searched refs:getLoopBody (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmhlo_control_flow_to_scf.cc134 forOp.getLoopBody().front().getOperations().splice( in MatchAndRewrite()
135 forOp.getLoopBody().front().getOperations().end(), in MatchAndRewrite()
138 b.setInsertionPointToStart(&forOp.getLoopBody().front()); in MatchAndRewrite()
174 forOp.getLoopBody().front().back().erase(); in MatchAndRewrite()
175 forOp.getLoopBody().front().back().erase(); in MatchAndRewrite()
176 b.setInsertionPointToEnd(&forOp.getLoopBody().front()); in MatchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/SCF/Transforms/
DStructuralTypeConversions.cpp54 rewriter.inlineRegionBefore(op.getLoopBody(), newOp.getLoopBody(), in matchAndRewrite()
55 newOp.getLoopBody().end()); in matchAndRewrite()
60 if (failed(rewriter.convertRegionTypes(&newOp.getLoopBody(), in matchAndRewrite()
/external/llvm-project/mlir/lib/Transforms/
DLoopInvariantCodeMotion.cpp79 auto &loopBody = looplike.getLoopBody(); in moveLoopInvariantCode()
/external/llvm-project/mlir/include/mlir/Interfaces/
DLoopLikeInterface.td39 "Region &", "getLoopBody"
/external/llvm-project/mlir/lib/Dialect/SCF/
DSCF.cpp234 Region &ForOp::getLoopBody() { return region(); } in getLoopBody() function in ForOp
278 regions.push_back(RegionSuccessor(&getLoopBody(), getRegionIterArgs())); in getSuccessorRegions()
284 regions.push_back(RegionSuccessor(&getLoopBody(), getRegionIterArgs())); in getSuccessorRegions()
556 replaceOpWithRegion(rewriter, op, op.getLoopBody(), blockArgs); in matchAndRewrite()
1000 Region &ParallelOp::getLoopBody() { return region(); } in getLoopBody() function in ParallelOp
/external/llvm-project/mlir/lib/Dialect/Tosa/IR/
DTosaOps.cpp63 Region &tosa::WhileOp::getLoopBody() { return body(); } in getLoopBody() function in tosa::WhileOp
/external/llvm-project/mlir/lib/Dialect/Async/Transforms/
DAsyncParallelFor.cpp201 for (auto &bodyOp : op.getLoopBody().getOps()) in matchAndRewrite()
/external/llvm-project/mlir/lib/Conversion/SCFToSPIRV/
DSCFToSPIRV.cpp158 body = rewriter.applySignatureConversion(&forOp.getLoopBody(), in matchAndRewrite()
/external/llvm-project/clang-tools-extra/clangd/
DXRefs.cpp840 const Stmt *getLoopBody(DynTypedNode N) { in getLoopBody() function
879 else if (getLoopBody(D)) in filterAndTraverse()
1026 if (const Stmt *LoopBody = getLoopBody(P->ASTNode)) { in relatedControlFlow()
/external/llvm-project/flang/lib/Optimizer/Dialect/
DFIROps.cpp753 mlir::Region &fir::IterWhileOp::getLoopBody() { return region(); } in getLoopBody() function in fir::IterWhileOp
939 mlir::Region &fir::LoopOp::getLoopBody() { return region(); } in getLoopBody() function in fir::LoopOp
/external/llvm-project/mlir/lib/Dialect/Affine/IR/
DAffineOps.cpp1759 Region &AffineForOp::getLoopBody() { return region(); } in getLoopBody() function in AffineForOp
2562 Region &AffineParallelOp::getLoopBody() { return region(); } in getLoopBody() function in AffineParallelOp
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2785 Region &WhileOp::getLoopBody() { return body(); } in getLoopBody() function in mlir::TFL::WhileOp
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2951 Region &WhileRegionOp::getLoopBody() { return body(); } in getLoopBody() function in mlir::TF::WhileRegionOp