Searched refs:transformed_while (Results 1 – 1 of 1) sorted by relevance
94 HloInstruction* transformed_while; in TEST_F() local95 FindOnlyWhileInstruction(entry_computation, &transformed_while); in TEST_F()98 EXPECT_THAT(transformed_while->while_body()->instructions(), in TEST_F()153 HloInstruction* transformed_while; in TEST_F() local154 FindOnlyWhileInstruction(entry_computation, &transformed_while); in TEST_F()165 EXPECT_THAT(transformed_while->while_body()->instructions(), in TEST_F()169 EXPECT_THAT(transformed_while->while_body()->instructions(), in TEST_F()394 HloInstruction* transformed_while; in TEST_F() local395 FindOnlyWhileInstruction(entry_computation, &transformed_while); in TEST_F()397 EXPECT_THAT(transformed_while->while_body()->instructions(), in TEST_F()[all …]