Searched refs:while_parent (Results 1 – 2 of 2) sorted by relevance
360 HloComputation* while_parent = old_while_instruction->parent(); in CreateNewWhileInit() local371 HloInstruction* new_buffer = while_parent->AddInstruction( in CreateNewWhileInit()386 HloInstruction* new_while_init = while_parent->AddInstruction( in CreateNewWhileInit()400 HloComputation* while_parent = new_while_instruction->parent(); in CreateSinkedAllReduces() local413 while_parent->AddInstruction(HloInstruction::CreateGetTupleElement( in CreateSinkedAllReduces()424 while_parent->AddInstruction(HloInstruction::CreateConvert( in CreateSinkedAllReduces()428 while_parent->AddInstruction(HloInstruction::CreateAllReduce( in CreateSinkedAllReduces()433 hlo_query::NextChannelId(*(while_parent->parent())), in CreateSinkedAllReduces()439 while_parent->AddInstruction(HloInstruction::CreateConvert( in CreateSinkedAllReduces()447 while_parent->AddInstruction(HloInstruction::CreateBinary( in CreateSinkedAllReduces()[all …]
535 if (auto while_parent = dyn_cast_or_null<WhileOp>(terminator->getParentOp())) in RewriteControlFlowTerminator() local536 if (terminator->getParentRegion() == &while_parent.cond()) return; in RewriteControlFlowTerminator()