Searched refs:while_instruction (Results 1 – 2 of 2) sorted by relevance
38 HloInstruction** while_instruction) { in FindOnlyWhileInstruction() argument39 *while_instruction = nullptr; in FindOnlyWhileInstruction()42 ASSERT_EQ(*while_instruction, nullptr); in FindOnlyWhileInstruction()43 *while_instruction = instr; in FindOnlyWhileInstruction()47 ASSERT_NE(*while_instruction, nullptr); in FindOnlyWhileInstruction()
1221 auto while_instruction = While(condition, body, init); in XLA_TEST_F() local1222 GetTupleElement(while_instruction, 3); in XLA_TEST_F()