Searched refs:GetInnerLoopBodyBasicBlock (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_loop_emitter.cc | 74 llvm_ir::SetToFirstInsertPoint(loop_nest.GetInnerLoopBodyBasicBlock(), b_); in EmitIndexAndSetExitBasicBlock()
|
| D | ir_emitter.cc | 736 SetToFirstInsertPoint(source_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
760 SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
1662 SetToFirstInsertPoint(reduction_loop_nest.GetInnerLoopBodyBasicBlock(), &b_); in EmitInnerLoopForVectorizedReduction()
1822 loop_nest.GetInnerLoopBodyBasicBlock()) { in EmitVectorizedReduce()
2046 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSlice()
2144 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in HandlePad()
2651 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in EmitFastConcatenate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_loop.h | 262 llvm::BasicBlock* GetInnerLoopBodyBasicBlock() { return inner_loop_body_bb_; } in GetInnerLoopBodyBasicBlock() function
|
| D | loop_emitter.cc | 162 llvm::BasicBlock* innermost_body_bb = loop_nest.GetInnerLoopBodyBasicBlock(); in EmitIndexAndSetExitBasicBlock()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 2839 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), b_); in EmitElementalReduceWindow()
2956 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), b()); in EmitElementalReduce()
3074 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), b_); in EmitConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2010 llvm_ir::SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), in EmitSelectAndScatter()
|