Searched refs:reduction_loop (Results 1 – 2 of 2) sorted by relevance
546 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in HandleDot() local553 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in HandleDot()554 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in HandleDot()569 reduction_loop->GetPreheaderBasicBlock() in HandleDot()576 TF_RET_CHECK(!reduction_loop->GetBodyBasicBlock()->empty()); in HandleDot()578 &*reduction_loop->GetBodyBasicBlock()->getFirstInsertionPt()); in HandleDot()622 SetToFirstInsertPoint(reduction_loop->GetExitBasicBlock(), &b_); in HandleDot()
447 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in EmitNaiveLlvmIrGemm() local457 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()460 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()470 llvm::Function* func = reduction_loop->GetPreheaderBasicBlock()->getParent(); in EmitNaiveLlvmIrGemm()478 llvm::BasicBlock* preheader_bb = reduction_loop->GetPreheaderBasicBlock(); in EmitNaiveLlvmIrGemm()488 SetToFirstInsertPoint(reduction_loop->GetBodyBasicBlock(), b_); in EmitNaiveLlvmIrGemm()517 SetToFirstInsertPoint(reduction_loop->GetExitBasicBlock(), b_); in EmitNaiveLlvmIrGemm()