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