Searched refs:xla_conditional (Results 1 – 1 of 1) sorted by relevance
197 const HloInstruction* xla_conditional, in IndicesToCopyForConditional() argument200 xla_conditional->shape())); in IndicesToCopyForConditional()207 CHECK_EQ(dataflow.GetValueSet(xla_conditional, index).values().size(), 1); in IndicesToCopyForConditional()209 auto value = dataflow.GetValueSet(xla_conditional, index).values()[0]; in IndicesToCopyForConditional()212 value->is_phi() && value->defining_instruction() == xla_conditional; in IndicesToCopyForConditional()