Searched refs:GrpcWorker (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.h | 39 class GrpcWorker : public Worker {
41 GrpcWorker(WorkerEnv* env, const ConfigProto& config);
70 std::unique_ptr<GrpcWorker> NewGrpcWorker(WorkerEnv* worker_env,
82 GrpcWorker* worker, ::grpc::ServerBuilder* builder,
|
| D | grpc_worker_service.cc | 102 GrpcWorker* worker, ::grpc::ServerBuilder* builder, in GrpcWorkerServiceThread()
356 GrpcWorker* const worker_ = nullptr; // Not owned.
370 GrpcWorkerService(GrpcWorker* worker, ::grpc::ServerBuilder* builder, in GrpcWorkerService()
422 GrpcWorker::GrpcWorker(WorkerEnv* worker_env, const ConfigProto& config) in GrpcWorker() function in tensorflow::GrpcWorker
433 void GrpcWorker::EnableResponseCache() { in EnableResponseCache()
441 void GrpcWorker::GrpcRecvTensorAsync(CallOptions* opts, in GrpcRecvTensorAsync()
591 void GrpcWorker::RecvBufAsync(CallOptions* opts, const RecvBufRequest* request, in RecvBufAsync()
709 void GrpcWorker::LoggingAsync(const LoggingRequest* request, in LoggingAsync()
736 void GrpcWorker::CleanupGraphAsync(const CleanupGraphRequest* request, in CleanupGraphAsync()
747 WorkerEnv* GrpcWorker::env() { return env_; } in env()
[all …]
|
| D | grpc_server_lib.h | 43 class GrpcWorker; variable
61 typedef std::function<std::unique_ptr<GrpcWorker>(WorkerEnv*,
170 GrpcWorker* worker_impl() const { return worker_impl_.get(); } in worker_impl()
210 std::unique_ptr<GrpcWorker> worker_impl_;
|