Searched refs:conditional_parent (Results 1 – 1 of 1) sorted by relevance
487 HloComputation* conditional_parent = conditional->parent(); in ConvertSpecialMove() local489 conditional_parent->AddInstruction(HloInstruction::CreateConditional( in ConvertSpecialMove()497 TF_CHECK_OK(conditional_parent->RemoveInstruction(conditional)); in ConvertSpecialMove()508 HloInstruction* gte = conditional_parent->AddInstruction( in ConvertSpecialMove()513 HloInstruction* hoisted = conditional_parent->AddInstruction( in ConvertSpecialMove()517 TF_CHECK_OK(conditional_parent->RemoveInstruction(gte_hoist)); in ConvertSpecialMove()538 HloComputation* conditional_parent = conditional->parent(); in MoveInstructionOut() local547 << conditional_parent->ToString(HloPrintOptions::Fingerprint()) in MoveInstructionOut()554 HloInstruction* gtr = conditional_parent->AddInstruction( in MoveInstructionOut()566 to_move_out[i], 0, conditional_parent, hoisted_instructions)); in MoveInstructionOut()[all …]