Searched refs:GrpcWorkerMethod (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service_impl.cc | 29 const char* GrpcWorkerMethodName(GrpcWorkerMethod id) { in GrpcWorkerMethodName()
31 case GrpcWorkerMethod::kGetStatus: in GrpcWorkerMethodName()
33 case GrpcWorkerMethod::kCreateWorkerSession: in GrpcWorkerMethodName()
35 case GrpcWorkerMethod::kDeleteWorkerSession: in GrpcWorkerMethodName()
37 case GrpcWorkerMethod::kRegisterGraph: in GrpcWorkerMethodName()
39 case GrpcWorkerMethod::kDeregisterGraph: in GrpcWorkerMethodName()
41 case GrpcWorkerMethod::kRunGraph: in GrpcWorkerMethodName()
43 case GrpcWorkerMethod::kCleanupGraph: in GrpcWorkerMethodName()
45 case GrpcWorkerMethod::kCleanupAll: in GrpcWorkerMethodName()
47 case GrpcWorkerMethod::kRecvTensor: in GrpcWorkerMethodName()
[all …]
|
| D | grpc_remote_worker.cc | 53 getstatus_(Method(GrpcWorkerMethod::kGetStatus)), in GrpcRemoteWorker()
54 createworkersession_(Method(GrpcWorkerMethod::kCreateWorkerSession)), in GrpcRemoteWorker()
55 deleteworkersession_(Method(GrpcWorkerMethod::kDeleteWorkerSession)), in GrpcRemoteWorker()
56 registergraph_(Method(GrpcWorkerMethod::kRegisterGraph)), in GrpcRemoteWorker()
57 deregistergraph_(Method(GrpcWorkerMethod::kDeregisterGraph)), in GrpcRemoteWorker()
58 rungraph_(Method(GrpcWorkerMethod::kRunGraph)), in GrpcRemoteWorker()
59 cleanupgraph_(Method(GrpcWorkerMethod::kCleanupGraph)), in GrpcRemoteWorker()
60 cleanupall_(Method(GrpcWorkerMethod::kCleanupAll)), in GrpcRemoteWorker()
61 recvtensor_(Method(GrpcWorkerMethod::kRecvTensor)), in GrpcRemoteWorker()
62 recvbuf_(Method(GrpcWorkerMethod::kRecvBuf)), in GrpcRemoteWorker()
[all …]
|
| D | grpc_worker_service_impl.h | 72 enum class GrpcWorkerMethod { enum
92 static_cast<int>(GrpcWorkerMethod::kMarkRecvFinished) + 1;
94 const char* GrpcWorkerMethodName(GrpcWorkerMethod id);
|
| D | grpc_worker_service.cc | 82 static_cast<int>(GrpcWorkerMethod::k##method), \
90 static_cast<int>(GrpcWorkerMethod::k##method), \
155 queue_depth_, static_cast<int>(GrpcWorkerMethod::kRecvTensor), in HandleRPCsLoop()
349 static_cast<int>(GrpcWorkerMethod::kRecvTensor), in EnqueueRecvTensorRequestRaw()
|