Searched refs:GetOuterLoopExitBasicBlock (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_loop_emitter.cc | 77 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
|
| D | ir_emitter.cc | 810 SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter()
830 SetToFirstInsertPoint(source_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter()
1599 SetToFirstInsertPoint(reduction_loop_nest.GetOuterLoopExitBasicBlock(), &b_); in EmitInnerLoopForVectorizedReduction()
1724 auto outermost_loop_exit_block = loop_nest.GetOuterLoopExitBasicBlock(); in EmitVectorizedReduce()
1963 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSlice()
2059 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in HandlePad()
2535 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in EmitFastConcatenate()
|
| D | dot_op_emitter.cc | 692 b_->SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in EmitNaiveLlvmIrGemm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_loop.h | 263 llvm::BasicBlock* GetOuterLoopExitBasicBlock() { return outer_loop_exit_bb_; } in GetOuterLoopExitBasicBlock() function
|
| D | loop_emitter.cc | 160 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 2703 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b_); in EmitElementalReduceWindow()
2794 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b()); in EmitElementalReduce()
2960 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b_); in EmitConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2391 llvm_ir::SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), in EmitSelectAndScatterFromMlir()
|