Searched refs:accumulator_addrs (Results 1 – 2 of 2) sorted by relevance
2723 std::vector<llvm::Value*> accumulator_addrs; in EmitElementalReduce() local2741 accumulator_addrs.push_back(accumulator_addr); in EmitElementalReduce()2774 for (llvm::Value* accum : accumulator_addrs) { in EmitElementalReduce()2792 Store(results[i], accumulator_addrs[i]); in EmitElementalReduce()2795 return EmitAccumResult(accumulator_addrs, accumulator_types, is_variadic); in EmitElementalReduce()2799 absl::Span<llvm::Value* const> accumulator_addrs, in EmitAccumResult() argument2801 TF_RET_CHECK(accumulator_addrs.size() == accumulator_types.size()); in EmitAccumResult()2806 for (int64 i = 0; i < accumulator_addrs.size(); i++) { in EmitAccumResult()2807 llvm::Value* accumulator_value = Load(accumulator_addrs[i]); in EmitAccumResult()2813 CHECK_EQ(accumulator_addrs.size(), 1); in EmitAccumResult()[all …]
188 absl::Span<llvm::Value* const> accumulator_addrs,