Searched refs:GrpcEagerServiceImpl (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/ |
| D | grpc_eager_service_impl.cc | 28 GrpcEagerServiceImpl::GrpcEagerServiceImpl( in GrpcEagerServiceImpl() function in tensorflow::eager::GrpcEagerServiceImpl
37 Status GrpcEagerServiceImpl::CreateMasterContext( in CreateMasterContext()
42 void GrpcEagerServiceImpl::HandleRPCsLoop() { in HandleRPCsLoop()
45 Call<GrpcEagerServiceImpl, grpc::EagerService::AsyncService, \ in HandleRPCsLoop()
49 &GrpcEagerServiceImpl::method##Handler, false); \ in HandleRPCsLoop()
61 ServerBidirectionalStreamingCall<GrpcEagerServiceImpl, in HandleRPCsLoop()
66 &GrpcEagerServiceImpl::StreamingEnqueueHandler); in HandleRPCsLoop()
76 GrpcCallTag<GrpcEagerServiceImpl>* callback_tag = in HandleRPCsLoop()
77 static_cast<GrpcCallTag<GrpcEagerServiceImpl>*>(tag); in HandleRPCsLoop()
88 void GrpcEagerServiceImpl::Shutdown() { in Shutdown()
|
| D | grpc_eager_service_impl.h | 32 class GrpcEagerServiceImpl : public AsyncServiceInterface {
35 using EagerCall = Call<GrpcEagerServiceImpl, grpc::EagerService::AsyncService,
39 ServerBidirectionalStreamingCall<GrpcEagerServiceImpl,
43 GrpcEagerServiceImpl(const WorkerEnv* env,
45 virtual ~GrpcEagerServiceImpl() {} in ~GrpcEagerServiceImpl()
61 Call<GrpcEagerServiceImpl, grpc::EagerService::AsyncService, \
65 &GrpcEagerServiceImpl::method##Handler, false); \
81 Call<GrpcEagerServiceImpl, grpc::EagerService::AsyncService, EnqueueRequest, in EnqueueHandler()
85 &GrpcEagerServiceImpl::EnqueueHandler, in EnqueueHandler()
102 Call<GrpcEagerServiceImpl, grpc::EagerService::AsyncService, in RunComponentFunctionHandler()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 251 eager_service_ = new eager::GrpcEagerServiceImpl(&worker_env_, &builder); in Init()
427 static_cast<eager::GrpcEagerServiceImpl*>(eager_service_); in AddMasterEagerContextToEagerService()
|