Home
last modified time | relevance | path

Searched refs:BranchOp (Results 1 – 25 of 38) sorted by relevance

12

/external/llvm-project/mlir/lib/Conversion/SCFToStandard/
DSCFToStandard.cpp317 rewriter.create<BranchOp>(loc, conditionBlock, loopCarried); in matchAndRewrite()
333 rewriter.create<BranchOp>(loc, conditionBlock, destOperands); in matchAndRewrite()
364 rewriter.create<BranchOp>(loc, remainingOpsBlock); in matchAndRewrite()
374 rewriter.create<BranchOp>(loc, continueBlock, thenTerminatorOperands); in matchAndRewrite()
388 rewriter.create<BranchOp>(loc, continueBlock, elseTerminatorOperands); in matchAndRewrite()
500 rewriter.create<BranchOp>(loc, before, whileOp.inits()); in matchAndRewrite()
513 rewriter.replaceOpWithNewOp<BranchOp>(yieldOp, before, yieldOp.results()); in matchAndRewrite()
555 rewriter.create<BranchOp>(whileOp.getLoc(), before, whileOp.inits()); in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/StandardOps/EDSC/
DIntrinsics.cpp15 BranchOp mlir::edsc::intrinsics::std_br(Block *block, ValueRange operands) { in std_br()
16 return OperationBuilder<BranchOp>(block, operands); in std_br()
/external/llvm-project/mlir/lib/Conversion/SCFToSPIRV/
DSCFToSPIRV.cpp170 rewriter.create<spirv::BranchOp>(loc, header, args); in matchAndRewrite()
189 rewriter.create<spirv::BranchOp>(loc, header, updatedIndVar); in matchAndRewrite()
232 rewriter.create<spirv::BranchOp>(loc, mergeBlock); in matchAndRewrite()
242 rewriter.create<spirv::BranchOp>(loc, mergeBlock); in matchAndRewrite()
279 cast<spirv::BranchOp>(rewriter.getInsertionBlock()->getTerminator()); in matchAndRewrite()
283 rewriter.create<spirv::BranchOp>(terminatorOp.getLoc(), br.getTarget(), in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_control_flow.cc56 builder->create<mlir::BranchOp>(loc, target_block, return_op.getOperands()); in ReplaceTerminators()
143 builder.create<mlir::BranchOp>(loc, cond_block, while_op.getOperand()); in LowerWhileOp()
197 builder.create<mlir::BranchOp>(loc, cond_block, return_op.getOperands()); in LowerWhileOp()
/external/llvm-project/mlir/lib/Dialect/GPU/Transforms/
DAllReduceLowering.cpp191 create<BranchOp>(block, ValueRange()); in getFactory()
199 rewriter.replaceOpWithNewOp<BranchOp>( in getFactory()
284 create<BranchOp>(continueBlock, thenOperands); in createIf()
288 create<BranchOp>(continueBlock, elseOperands); in createIf()
DKernelOutlining.cpp187 builder.create<BranchOp>(loc, clonedLaunchOpEntry); in outlineKernelFuncImpl()
/external/llvm-project/mlir/test/Dialect/Standard/
Dcanonicalize-cf.mlir3 /// Test the folding of BranchOp.
15 /// Test that pass-through successors of BranchOp get folded.
86 /// Test the compound folding of BranchOp and CondBranchOp.
/external/llvm-project/mlir/lib/Dialect/StandardOps/IR/
DOps.cpp74 builder.create<BranchOp>(op->getLoc(), newDest, returnOp.getOperands()); in handleTerminator()
574 BranchOp successorBranch = dyn_cast<BranchOp>(successor->getTerminator()); in collapseBranch()
613 struct SimplifyBrToBlockWithSinglePred : public OpRewritePattern<BranchOp> {
614 using OpRewritePattern<BranchOp>::OpRewritePattern;
616 LogicalResult matchAndRewrite(BranchOp op, in matchAndRewrite()
637 struct SimplifyPassThroughBr : public OpRewritePattern<BranchOp> {
638 using OpRewritePattern<BranchOp>::OpRewritePattern;
640 LogicalResult matchAndRewrite(BranchOp op, in matchAndRewrite()
653 rewriter.replaceOpWithNewOp<BranchOp>(op, dest, destOperands); in matchAndRewrite()
659 Block *BranchOp::getDest() { return getSuccessor(); } in getDest()
[all …]
/external/llvm-project/mlir/lib/Dialect/SPIRV/
DSPIRVCanonicalization.cpp383 dyn_cast<spirv::BranchOp>(*std::next(trueBlock->begin())); in canCanonicalizeSelection()
386 dyn_cast<spirv::BranchOp>(*std::next(falseBlock->begin())); in canCanonicalizeSelection()
/external/llvm-project/mlir/include/mlir/Dialect/StandardOps/EDSC/
DIntrinsics.h57 BranchOp std_br(Block *block, ValueRange operands);
/external/llvm/lib/Target/Mips/
DMipsSEISelLowering.h84 unsigned BranchOp) const;
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/Mips/
DMipsSEISelLowering.h95 unsigned BranchOp) const;
/external/llvm-project/llvm/lib/Target/Mips/
DMipsSEISelLowering.h95 unsigned BranchOp) const;
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/AVR/
DAVRInstrInfo.cpp532 bool AVRInstrInfo::isBranchOffsetInRange(unsigned BranchOp, in isBranchOffsetInRange() argument
535 switch (BranchOp) { in isBranchOffsetInRange()
/external/llvm-project/llvm/lib/Target/AVR/
DAVRInstrInfo.cpp532 bool AVRInstrInfo::isBranchOffsetInRange(unsigned BranchOp, in isBranchOffsetInRange() argument
535 switch (BranchOp) { in isBranchOffsetInRange()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfunctional_control_flow_to_cfg.cc107 builder->create<BranchOp>(loc, block, operands); in JumpToBlock()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/RISCV/
DRISCVInstrInfo.cpp432 bool RISCVInstrInfo::isBranchOffsetInRange(unsigned BranchOp, in isBranchOffsetInRange() argument
437 switch (BranchOp) { in isBranchOffsetInRange()
/external/llvm-project/mlir/lib/Dialect/SPIRV/Serialization/
DDeserializer.cpp1782 opBuilder.create<spirv::BranchOp>(loc, target); in processBranch()
2120 builder.create<spirv::BranchOp>(location, mapper.lookupOrNull(headerBlock), in structurizeImpl()
2177 builder.create<spirv::BranchOp>(location, mergeBlock); in structurizeImpl()
2222 if (auto branchOp = dyn_cast<spirv::BranchOp>(op)) { in wireUpBlockArgument()
2224 opBuilder.create<spirv::BranchOp>(branchOp.getLoc(), branchOp.getTarget(), in wireUpBlockArgument()
DSerializer.cpp347 LogicalResult processBranchOp(spirv::BranchOp branchOp);
1658 if (auto branchOp = dyn_cast<spirv::BranchOp>(terminator)) { in emitPhiForBlockArguments()
1834 LogicalResult Serializer::processBranchOp(spirv::BranchOp branchOp) { in processBranchOp()
1907 .Case([&](spirv::BranchOp op) { return processBranchOp(op); }) in processOperation()
/external/llvm-project/llvm/lib/Target/RISCV/
DRISCVInstrInfo.cpp448 bool RISCVInstrInfo::isBranchOffsetInRange(unsigned BranchOp, in isBranchOffsetInRange() argument
454 switch (BranchOp) { in isBranchOffsetInRange()
/external/llvm-project/mlir/lib/Conversion/AsyncToLLVM/
DAsyncToLLVM.cpp361 builder.create<BranchOp>(loc, suspendBlock); in setupCoroMachinery()
380 builder.create<BranchOp>(loc, cleanupBlock); in setupCoroMachinery()
/external/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/
DConvertSPIRVToLLVM.cpp520 class BranchConversionPattern : public SPIRVToLLVMConversion<spirv::BranchOp> {
522 using SPIRVToLLVMConversion<spirv::BranchOp>::SPIRVToLLVMConversion;
525 matchAndRewrite(spirv::BranchOp branchOp, ArrayRef<Value> operands, in matchAndRewrite()
1060 auto brOp = dyn_cast<spirv::BranchOp>(entryBlock->getOperations().front()); in matchAndRewrite()
/external/llvm-project/mlir/lib/Rewrite/
DByteCode.cpp199 void generate(pdl_interp::BranchOp op, ByteCodeWriter &writer);
491 pdl_interp::AreEqualOp, pdl_interp::BranchOp, in generate()
531 void Generator::generate(pdl_interp::BranchOp op, ByteCodeWriter &writer) { in generate()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dpasses.cc295 kernel_builder.create<mlir::BranchOp>( in runOnFunction()
/external/llvm-project/mlir/test/Rewrite/
Dpdl-bytecode.mlir112 // pdl_interp::BranchOp

12