Lines Matching defs:while_hlo
641 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile(
644 return while_hlo;
649 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile(
652 return while_hlo;
755 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile(
758 return while_hlo;
787 auto while_hlo = BuildWhileInstruction(condition, body);
796 EXPECT_THAT(while_hlo->operand(0),
815 auto while_hlo = BuildWhileInstruction(condition, body);
832 while_hlo->while_body()->root_instruction(),
837 EXPECT_THAT(while_hlo->operand(0),
1046 auto while_hlo = BuildWhileInstruction_InitPointsToConstant();
1049 EXPECT_EQ(CountCopies(*while_hlo->while_body()), 0);
1052 EXPECT_THAT(while_hlo->operand(0),
1063 auto while_hlo = BuildWhileInstruction_InitPointsToParameter();
1066 EXPECT_EQ(CountCopies(*while_hlo->while_body()), 0);
1069 EXPECT_THAT(while_hlo->operand(0),
1098 auto while_hlo = BuildWhileInstruction_InitPointsToAmbiguous();
1105 EXPECT_THAT(while_hlo->operand(0),
1113 EXPECT_EQ(CountCopies(*while_hlo->while_body()), 1);
1114 if (while_hlo->while_body()
1120 while_hlo->while_body()->root_instruction(),
1124 while_hlo->while_body()->root_instruction(),
1153 auto while_hlo = BuildWhileInstruction_InitPointsToNonDistinct();
1161 if (while_hlo->operand(0)->operand(1)->operand(0)->opcode() ==
1164 while_hlo->operand(0),
1169 while_hlo->operand(0),
1177 EXPECT_EQ(CountCopies(*while_hlo->while_body()), 1);
1178 if (while_hlo->while_body()
1184 while_hlo->while_body()->root_instruction(),
1188 while_hlo->while_body()->root_instruction(),
1203 auto while_hlo = BuildWhileInstruction_InitPointsToInterfering();
1207 EXPECT_EQ(CountCopies(*while_hlo->while_body()), 0);
1209 EXPECT_THAT(while_hlo->operand(0),
