Searched refs:reduce_comp (Results 1 – 2 of 2) sorted by relevance
116 HloComputation* reduce_comp = hlo.called_computations()[0]; in EffectiveOperandPrecisionIsOutputPrecision() local117 for (HloInstruction* inst : reduce_comp->instructions()) { in EffectiveOperandPrecisionIsOutputPrecision()
1620 auto reduce_comp = consumer->to_apply(); in Propagate() local1627 reduce_comp)); in Propagate()