Searched refs:extra_output_gens (Results 1 – 2 of 2) sorted by relevance
213 extra_output_gens);
709 extra_output_gens) { in EmitExtraOutputsForReduce() argument710 for (int i = 0; i != extra_output_gens.size(); ++i) { in EmitExtraOutputsForReduce()712 GetIrArray(*unnested_hlo, *unnested_hlo, extra_output_gens[i].second) in EmitExtraOutputsForReduce()716 extra_output_gens[i].first(index)); in EmitExtraOutputsForReduce()2882 extra_output_gens; in EmitTileElementForReduction() local2904 extra_output_gens.emplace_back(fused_emitter.GetGenerator(inst), in EmitTileElementForReduction()2942 EmitExtraOutputsForReduce(unnested_hlo, input_index, extra_output_gens)); in EmitTileElementForReduction()