Searched refs:partial_result_addresses (Results 1 – 2 of 2) sorted by relevance
2666 AddressVector* partial_result_addresses = in EmitPrologueForOneReduction() local2671 partial_result_addresses->push_back(partial_result_address); in EmitPrologueForOneReduction()2751 absl::Span<llvm::AllocaInst* const> partial_result_addresses) { in EmitFullWarpShuffleDownLoopForAllReduces() argument2755 partial_result_addresses[i]->getType()->getElementType(); in EmitFullWarpShuffleDownLoopForAllReduces()2767 Load(convert_pointer_for_shuffle(partial_result_addresses[i]), in EmitFullWarpShuffleDownLoopForAllReduces()2772 *reducers[i], {partial_result_addresses[i], result_from_other_lane}, in EmitFullWarpShuffleDownLoopForAllReduces()2773 partial_result_addresses[i])); in EmitFullWarpShuffleDownLoopForAllReduces()2783 absl::Span<llvm::AllocaInst* const> partial_result_addresses = in EmitEpilogueForReduction() local2792 partial_result_addresses); in EmitEpilogueForReduction()2841 InBoundsGEP(partial_result_addresses[i], {b_.getInt32(j)})}, in EmitEpilogueForReduction()[all …]
290 absl::Span<llvm::AllocaInst* const> partial_result_addresses);