Searched refs:GetFirstReduceInstruction (Results 1 – 1 of 1) sorted by relevance
2630 const HloInstruction* GetFirstReduceInstruction( in GetFirstReduceInstruction() function2918 GetFirstReduceInstruction(output_instructions)->operand(0)->shape()); in EmitTileElementForReduction()3738 GetFirstReduceInstruction(output_instructions); in EmitReductionToVector()