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()
2033 blocking_op_state->AwaitState(1); in TestSessionInterOpThreadsImpl()2035 blocking_op_state->AwaitState(3); in TestSessionInterOpThreadsImpl()2046 blocking_op_state->AwaitState(1); // Wait for the blocking op to Compute. in TestSessionInterOpThreadsImpl()