Home
last modified time | relevance | path

Searched refs:GetInnerLoopBodyBasicBlock (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc74 llvm_ir::SetToFirstInsertPoint(loop_nest.GetInnerLoopBodyBasicBlock(), b_); in EmitIndexAndSetExitBasicBlock()
Dir_emitter.cc713 SetToFirstInsertPoint(source_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
736 SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
1564 SetToFirstInsertPoint(reduction_loop_nest.GetInnerLoopBodyBasicBlock(), &b_); in EmitInnerLoopForVectorizedReduction()
1720 loop_nest.GetInnerLoopBodyBasicBlock()) { in EmitVectorizedReduce()
1935 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSlice()
2033 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in HandlePad()
2490 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in EmitFastConcatenate()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_loop.h264 llvm::BasicBlock* GetInnerLoopBodyBasicBlock() { return inner_loop_body_bb_; } in GetInnerLoopBodyBasicBlock() function
Dloop_emitter.cc155 llvm::BasicBlock* innermost_body_bb = loop_nest.GetInnerLoopBodyBasicBlock(); in EmitIndexAndSetExitBasicBlock()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2642 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), b_); in EmitElementalReduceWindow()
2755 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), b()); in EmitElementalReduce()
2867 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), b_); in EmitConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2298 llvm_ir::SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), in EmitSelectAndScatterFromMlir()