Searched refs:TerminatorOp (Results 1 – 16 of 16) sorted by relevance
| /external/llvm-project/mlir/test/lib/Dialect/Test/ |
| D | TestPatterns.cpp | 257 rewriter.create<TerminatorOp>(op->getLoc()); in matchAndRewrite()
276 rewriter.create<TerminatorOp>(op->getLoc()); in matchAndRewrite()
576 TerminatorOp>(); in runOnOperation()
906 TerminatorOp innerTerminator = in matchAndRewrite()
907 cast<TerminatorOp>(innerBlock.getTerminator()); in matchAndRewrite()
926 target.addLegalOp<FuncOp, ModuleOp, ModuleTerminatorOp, TerminatorOp, in runOnOperation()
|
| D | TestOps.td | 107 [SymbolTable, SingleBlockImplicitTerminator<"TerminatorOp">]> {
485 def TerminatorOp : TEST_Op<"finish", [Terminator]>;
487 [SingleBlockImplicitTerminator<"TerminatorOp">]> {
|
| /external/llvm-project/mlir/lib/Conversion/OpenMPToLLVM/ |
| D | OpenMPToLLVM.cpp | 74 target.addLegalOp<omp::TerminatorOp, omp::TaskyieldOp, omp::FlushOp, in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_ops.td | 205 // The region `body` may return lmhlo.TerminatorOp or mhlo.ReturnOp. We are
206 // moving towards mhlo.ReturnOp, but some code that needs cleanup still assumes lmhlo.TerminatorOp.
207 // TODO(timshen): cleanup lmhlo.TerminatorOp.
247 SingleBlockImplicitTerminator<"TerminatorOp">
678 def FusionOp : LHLO_Op<"fusion", [SingleBlockImplicitTerminator<"TerminatorOp">]> {
737 def TerminatorOp :
|
| /external/llvm-project/mlir/lib/Conversion/SCFToOpenMP/ |
| D | SCFToOpenMP.cpp | 78 builder.create<omp::TerminatorOp>(parallelOp.getLoc()); in insertOpenMPParallel()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_gpu.cc | 164 rewriter.create<mlir::gpu::TerminatorOp>(loc); in matchAndRewrite()
|
| D | hlo_legalize_to_lhlo.cc | 480 rewriter.replaceOpWithNewOp<lmhlo::TerminatorOp>(op); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/SCFToGPU/ |
| D | SCFToGPU.cpp | 230 builder.create<gpu::TerminatorOp>(terminatorLoc, llvm::None); in createLaunch()
581 rewriter.create<gpu::TerminatorOp>(loc); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | KernelOutlining.cpp | 189 outlinedFunc.walk([](gpu::TerminatorOp op) { in outlineKernelFuncImpl()
|
| /external/llvm-project/flang/lib/Lower/ |
| D | OpenMP.cpp | 58 firOpBuilder.create<mlir::omp::TerminatorOp>(loc); in createBodyOfOp()
|
| D | OpenACC.cpp | 514 createRegionOp<mlir::acc::DataOp, mlir::acc::TerminatorOp>( in genACCDataOp()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/IR/ |
| D | GPUDialect.cpp | 311 if (!isa<gpu::TerminatorOp>(&block.back())) { in verify()
314 .append("expected '", gpu::TerminatorOp::getOperationName(), in verify()
|
| /external/llvm-project/mlir/include/mlir/Dialect/OpenMP/ |
| D | OpenMPOps.td | 103 def TerminatorOp : OpenMP_Op<"terminator", [Terminator]> {
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_list_ops_decomposition.cc | 130 template <class TerminatorOp>
146 .create<TerminatorOp>(old_terminator->getLoc(), new_outputs); in AddTensorListSizesToTerminator()
|
| /external/llvm-project/mlir/lib/Target/LLVMIR/ |
| D | ModuleTranslation.cpp | 551 .Case([&](omp::TerminatorOp) { in convertOmpOperation() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 263 if (mlir::isa<mlir::lmhlo::TerminatorOp, mlir::mhlo::ReturnOp, in MayPreventVectorization()
1823 mlir::lmhlo::TerminatorOp, mlir::mhlo::ReturnOp>(op)) { in EmitLoopFusionFromMlir()
|