Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dparallel_loop_emitter.cc77 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
H A Dir_emitter.cc841 SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter()
865 SetToFirstInsertPoint(source_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter()
1699 SetToFirstInsertPoint(reduction_loop_nest.GetOuterLoopExitBasicBlock(), &b_); in EmitInnerLoopForVectorizedReduction()
1826 auto outermost_loop_exit_block = loop_nest.GetOuterLoopExitBasicBlock(); in EmitVectorizedReduce()
2074 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSlice()
2170 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in HandlePad()
2697 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in EmitFastConcatenate()
H A Ddot_op_emitter.cc756 b_->SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in EmitNaiveLlvmIrGemm()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dllvm_loop.h261 llvm::BasicBlock* GetOuterLoopExitBasicBlock() { return outer_loop_exit_bb_; } in GetOuterLoopExitBasicBlock() function
H A Dloop_emitter.cc167 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Delemental_ir_emitter.cc2904 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b_); in EmitElementalReduceWindow()
2996 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b()); in EmitElementalReduce()
3174 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b_); in EmitConvolution()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.cc2105 llvm_ir::SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), in EmitSelectAndScatter()