Home
last modified time | relevance | path

Searched refs:GetExitBasicBlock (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_support_library.cc54 llvm_ir::SetToLastInsertPoint(loop->GetExitBasicBlock(), b_); in ForWithStatus()
Dllvm_loop.h129 llvm::BasicBlock* GetExitBasicBlock() const { return exit_bb_; } in GetExitBasicBlock() function
Dllvm_loop.cc202 outer_loop_exit_bb_ = loop->GetExitBasicBlock(); in AddLoop()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc622 SetToFirstInsertPoint(reduction_loop->GetExitBasicBlock(), &b_); in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc517 SetToFirstInsertPoint(reduction_loop->GetExitBasicBlock(), b_); in EmitNaiveLlvmIrGemm()
Dir_emitter.cc1772 if (auto exit_terminator = loop->GetExitBasicBlock()->getTerminator()) { in EmitVectorizedReduce()
1777 b_.SetInsertPoint(loop->GetExitBasicBlock()); in EmitVectorizedReduce()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2198 SetToFirstInsertPoint(inner_loop->GetExitBasicBlock(), b_); in EmitElementalDot()