Searched refs:extra_output_gens (Results 1 – 2 of 2) sorted by relevance
369 extra_output_gens);
2121 extra_output_gens) { in EmitExtraOutputsForReduce() argument2125 for (int i = 0; i < extra_output_gens.size(); ++i) { in EmitExtraOutputsForReduce()2127 extra_output_gens[i].first(index)); in EmitExtraOutputsForReduce()2130 for (int i = 0; i < extra_output_gens.size(); ++i) { in EmitExtraOutputsForReduce()2131 result_ir_arrays[extra_output_gens[i].second].EmitWriteArrayElement( in EmitExtraOutputsForReduce()4607 std::vector<std::pair<llvm_ir::ElementGenerator, int>> extra_output_gens; in EmitTileElementForReduction() local4622 extra_output_gens.emplace_back(*fused_emitter->GetGenerator(hlo), in EmitTileElementForReduction()4666 /*use_linear_index=*/num_partial_results == 1, extra_output_gens)); in EmitTileElementForReduction()