Searched refs:ComputeMappingSchemeAndReductionKind (Results 1 – 2 of 2) sorted by relevance
226 ComputeMappingSchemeAndReductionKind(const HloInstruction* unnested_hlo,
3649 IrEmitterUnnested::ComputeMappingSchemeAndReductionKind( in ComputeMappingSchemeAndReductionKind() function in xla::gpu::IrEmitterUnnested3774 ComputeMappingSchemeAndReductionKind(unnested_hlo, first_reduce); in EmitReductionToVector()