Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_support_library.cc49 b_->SetInsertPoint(&loop->GetBodyBasicBlock()->back()); in ForWithStatus()
Dllvm_loop.h128 llvm::BasicBlock* GetBodyBasicBlock() const { return body_bb_; } in GetBodyBasicBlock() function
Dllvm_loop.cc205 inner_loop_body_bb_ = loop->GetBodyBasicBlock(); in AddLoop()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc576 TF_RET_CHECK(!reduction_loop->GetBodyBasicBlock()->empty()); in HandleDot()
578 &*reduction_loop->GetBodyBasicBlock()->getFirstInsertionPt()); in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc488 SetToFirstInsertPoint(reduction_loop->GetBodyBasicBlock(), b_); in EmitNaiveLlvmIrGemm()
Dir_emitter.cc1755 SetToFirstInsertPoint(loop->GetBodyBasicBlock(), &b_); in EmitVectorizedReduce()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2144 SetToFirstInsertPoint(inner_loop->GetBodyBasicBlock(), b_); in EmitElementalDot()