Searched refs:AwaitState (Results 1 – 4 of 4) sorted by relevance
61 void BlockingOpState::AwaitState(int awaiting_state) { in AwaitState() function in tensorflow::test::function::BlockingOpState84 blocking_op_state->AwaitState(2); in Compute()
30 void AwaitState(int awaiting_state);
240 blocking_op_state->AwaitState(1); in TEST_F()244 blocking_op_state->AwaitState(3); in TEST_F()
2062 blocking_op_state->AwaitState(1); in TestSessionInterOpThreadsImpl()2064 blocking_op_state->AwaitState(3); in TestSessionInterOpThreadsImpl()2075 blocking_op_state->AwaitState(1); // Wait for the blocking op to Compute. in TestSessionInterOpThreadsImpl()