Searched refs:EmitExtraOutputsForReduce (Results 1 – 2 of 2) sorted by relevance
210 Status EmitExtraOutputsForReduce(
706 Status IrEmitterUnnested::EmitExtraOutputsForReduce( in EmitExtraOutputsForReduce() function in xla::gpu::IrEmitterUnnested2942 EmitExtraOutputsForReduce(unnested_hlo, input_index, extra_output_gens)); in EmitTileElementForReduction()