Searched refs:GetInnerLoopBodyBasicBlock (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_loop_emitter.cc | 68 llvm_ir::SetToFirstInsertPoint(loop_nest.GetInnerLoopBodyBasicBlock(), b_); in EmitIndexAndSetExitBasicBlock()
|
| D | ir_emitter.cc | 666 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in EmitElementalReduceWindow()
794 SetToFirstInsertPoint(source_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
817 SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
996 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in EmitElementalConvolution()
1599 SetToFirstInsertPoint(reduction_loop_nest.GetInnerLoopBodyBasicBlock(), &b_); in EmitInnerLoopForVectorizedReduction()
1740 loop_nest.GetInnerLoopBodyBasicBlock()) { in EmitVectorizedReduce()
1842 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in EmitElementalReduce()
2021 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSlice()
2119 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in HandlePad()
2429 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in EmitFastConcatenate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 114 llvm::BasicBlock* innermost_body_bb = loop_nest.GetInnerLoopBodyBasicBlock(); in EmitIndexAndSetExitBasicBlock()
|
| D | llvm_loop.h | 264 llvm::BasicBlock* GetInnerLoopBodyBasicBlock() { return inner_loop_body_bb_; } in GetInnerLoopBodyBasicBlock() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.cc | 386 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), b_); in MakeElementGenerator()
473 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), b()); in MakeElementGenerator()
|
| D | ir_emitter.cc | 693 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleReduce()
|
| D | ir_emitter_unnested.cc | 860 llvm_ir::SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), in HandleSelectAndScatter()
|