Searched refs:partial_result_addresses (Results 1 – 2 of 2) sorted by relevance
631 absl::Span<llvm::AllocaInst* const> partial_result_addresses,
4255 AddressVector* partial_result_addresses = in EmitPrologueForReduction() local4261 partial_result_addresses->push_back(partial_result_address); in EmitPrologueForReduction()4320 absl::Span<llvm::AllocaInst* const> partial_result_addresses, in EmitFullWarpShuffleDownLoopForAllReduces() argument4322 CHECK_EQ(reducers.size(), partial_result_addresses.size()); in EmitFullWarpShuffleDownLoopForAllReduces()4325 reducers[i], partial_result_addresses[i]->getType()->getElementType(), in EmitFullWarpShuffleDownLoopForAllReduces()4326 partial_result_addresses[i], threads_per_block); in EmitFullWarpShuffleDownLoopForAllReduces()4406 absl::Span<llvm::AllocaInst* const> partial_result_addresses = in EmitEpilogueForReduction() local4458 partial_result_addresses[i], {constant(j)}, "current_output"); in EmitEpilogueForReduction()4478 partial_result_addresses[i]->getType()->getElementType(); in EmitEpilogueForReduction()