Searched refs:GrpcWorker (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.h | 38 class GrpcWorker : public Worker {
40 GrpcWorker(WorkerEnv* env, const ConfigProto& config);
61 std::unique_ptr<GrpcWorker> NewGrpcWorker(WorkerEnv* worker_env,
75 GrpcWorker* worker, ::grpc::ServerBuilder* builder,
|
| D | grpc_worker_service.cc | 101 GrpcWorker* worker, ::grpc::ServerBuilder* builder, in GrpcWorkerServiceThread()
362 GrpcWorker* const worker_ = nullptr; // Not owned.
376 GrpcWorkerService(GrpcWorker* worker, ::grpc::ServerBuilder* builder, in GrpcWorkerService()
433 GrpcWorker::GrpcWorker(WorkerEnv* worker_env, const ConfigProto& config) in GrpcWorker() function in tensorflow::GrpcWorker
444 void GrpcWorker::GrpcRecvTensorAsync(CallOptions* opts, in GrpcRecvTensorAsync()
555 void GrpcWorker::RecvBufAsync(CallOptions* opts, const RecvBufRequest* request, in RecvBufAsync()
623 void GrpcWorker::LoggingAsync(const LoggingRequest* request, in LoggingAsync()
650 WorkerEnv* GrpcWorker::env() { return env_; } in env()
652 std::unique_ptr<GrpcWorker> NewGrpcWorker(WorkerEnv* env, in NewGrpcWorker()
654 return std::unique_ptr<GrpcWorker>(new GrpcWorker(env, config)); in NewGrpcWorker()
[all …]
|
| D | grpc_server_lib.h | 41 class GrpcWorker; variable
59 typedef std::function<std::unique_ptr<GrpcWorker>(WorkerEnv*,
157 std::unique_ptr<GrpcWorker> worker_impl_;
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_worker.h | 26 class GdrWorker : public GrpcWorker {
|
| D | gdr_worker.cc | 43 : GrpcWorker(worker_env, config), in GdrWorker()
|