Searched refs:GetOuterLoopExitBasicBlock (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | parallel_loop_emitter.cc | 71 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
|
D | ir_emitter.cc | 724 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in EmitElementalReduceWindow() 891 SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter() 911 SetToFirstInsertPoint(source_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter() 1089 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in EmitElementalConvolution() 1634 SetToFirstInsertPoint(reduction_loop_nest.GetOuterLoopExitBasicBlock(), &b_); in EmitInnerLoopForVectorizedReduction() 1744 auto outermost_loop_exit_block = loop_nest.GetOuterLoopExitBasicBlock(); in EmitVectorizedReduce() 1868 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in EmitElementalReduce() 2049 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSlice() 2145 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in HandlePad() 2472 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in EmitFastConcatenate()
|
D | dot_op_emitter.cc | 543 b_->SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in EmitNaiveLlvmIrGemm()
|
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
D | loop_emitter.cc | 119 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
|
D | llvm_loop.h | 263 llvm::BasicBlock* GetOuterLoopExitBasicBlock() { return outer_loop_exit_bb_; } in GetOuterLoopExitBasicBlock() function
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | elemental_ir_emitter.cc | 437 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b_); in MakeElementGenerator() 482 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b()); in MakeElementGenerator()
|
D | ir_emitter.cc | 632 b_.SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in HandleDot() 717 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in HandleReduce()
|
D | ir_emitter_unnested.cc | 941 llvm_ir::SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), in HandleSelectAndScatter()
|