Searched refs:new_while_op (Results 1 – 2 of 2) sorted by relevance
411 HloInstruction* new_while_op = in TEST_F() local419 ShapeUtil::Equal(new_while_op->shape(), in TEST_F()421 << ShapeUtil::HumanString(new_while_op->shape()); in TEST_F()423 new_while_op->while_body()->root_instruction(), in TEST_F()429 EXPECT_THAT(new_while_op->while_condition()->root_instruction(), in TEST_F()
263 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in TryRemoveDeadWhileParams() local293 new_while_op->shape().tuple_shapes(gte_idx), new_while_op, in TryRemoveDeadWhileParams()1058 TF_ASSIGN_OR_RETURN(auto* new_while_op, in Run()1060 if (new_while_op) { in Run()1061 while_op = new_while_op; in Run()