Searched refs:outer_while (Results 1 – 2 of 2) sorted by relevance
249 auto outer_while = xla::While(outer_loop_cond, outer_loop_body, init); in Compile() local250 auto output_indices = xla::GetTupleElement(outer_while, 1); in Compile()
865 HloInstruction* outer_while = in TEST_F() local884 TF_CHECK_OK(while_hlo->parent()->ReplaceInstruction(while_hlo, outer_while)); in TEST_F()933 HloInstruction* outer_while = in TEST_F() local944 TF_CHECK_OK(while_hlo->parent()->ReplaceInstruction(while_hlo, outer_while)); in TEST_F()993 HloInstruction* outer_while = in TEST_F() local1012 TF_CHECK_OK(while_hlo->parent()->ReplaceInstruction(while_hlo, outer_while)); in TEST_F()