Searched refs:outer_while (Results 1 – 2 of 2) sorted by relevance
783 auto* outer_while = in TEST_F() local785 ASSERT_NE(outer_while, nullptr); in TEST_F()787 CheckWhileBound(outer_while, kMaxLoopCount); in TEST_F()788 auto* while_body = outer_while->while_body(); in TEST_F()
769 HloInstruction* outer_while = in TEST_F() local788 TF_CHECK_OK(while_hlo->parent()->ReplaceInstruction(while_hlo, outer_while)); in TEST_F()836 HloInstruction* outer_while = in TEST_F() local847 TF_CHECK_OK(while_hlo->parent()->ReplaceInstruction(while_hlo, outer_while)); in TEST_F()895 HloInstruction* outer_while = in TEST_F() local914 TF_CHECK_OK(while_hlo->parent()->ReplaceInstruction(while_hlo, outer_while)); in TEST_F()