Searched refs:enqueue_request (Results 1 – 1 of 1) sorted by relevance
492 EnqueueRequest enqueue_request; in TestFunction() local493 enqueue_request.set_context_id(context_id); in TestFunction()494 *enqueue_request.add_queue()->mutable_register_function() = register_op; in TestFunction()497 TF_ASSERT_OK(eager_service_impl.Enqueue(nullptr, &enqueue_request, in TestFunction()591 EnqueueRequest enqueue_request; in TestComponentFunction() local592 enqueue_request.set_context_id(context_id); in TestComponentFunction()593 *enqueue_request.add_queue()->mutable_register_function() = register_op; in TestComponentFunction()595 TF_ASSERT_OK(eager_service_impl.Enqueue(nullptr, &enqueue_request, in TestComponentFunction()