Searched refs:nested_while (Results 1 – 2 of 2) sorted by relevance
737 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local759 analysis.GetUniqueBufferAt(nested_while, /*index=*/{0})); in TEST_F()766 analysis.GetUniqueBufferAt(nested_while, /*index=*/{1})); in TEST_F()
697 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P() local725 EXPECT_FALSE(analysis.ValueIsDefinedAt(nested_while, /*index=*/{0})); in TEST_P()729 EXPECT_TRUE(analysis.ValueIsDefinedAt(nested_while, /*index=*/{1})); in TEST_P()731 analysis.GetValueDefinedAt(nested_while, /*index=*/{1}).is_phi()); in TEST_P()745 EXPECT_THAT(HloValuesAt(nested_while, /*index=*/{0}), in TEST_P()747 EXPECT_THAT(HloValuesAt(nested_while, /*index=*/{1}), in TEST_P()