Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc590 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in EmitNaiveLlvmIrGemm() local
600 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()
603 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()
613 llvm::Function* func = reduction_loop->GetPreheaderBasicBlock()->getParent(); in EmitNaiveLlvmIrGemm()
621 llvm::BasicBlock* preheader_bb = reduction_loop->GetPreheaderBasicBlock(); in EmitNaiveLlvmIrGemm()
631 SetToFirstInsertPoint(reduction_loop->GetBodyBasicBlock(), b_); in EmitNaiveLlvmIrGemm()
666 SetToFirstInsertPoint(reduction_loop->GetExitBasicBlock(), b_); in EmitNaiveLlvmIrGemm()