Home
last modified time | relevance | path

Searched refs:GetOuterLoopExitBasicBlock (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc71 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
Dir_emitter.cc724 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()
Ddot_op_emitter.cc543 b_->SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in EmitNaiveLlvmIrGemm()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dloop_emitter.cc119 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
Dllvm_loop.h263 llvm::BasicBlock* GetOuterLoopExitBasicBlock() { return outer_loop_exit_bb_; } in GetOuterLoopExitBasicBlock() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc437 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b_); in MakeElementGenerator()
482 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b()); in MakeElementGenerator()
Dir_emitter.cc632 b_.SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in HandleDot()
717 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in HandleReduce()
Dir_emitter_unnested.cc941 llvm_ir::SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), in HandleSelectAndScatter()