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 | 54 getstatus_(Method(GrpcWorkerMethod::kGetStatus)), in GrpcRemoteWorker() 55 createworkersession_(Method(GrpcWorkerMethod::kCreateWorkerSession)), in GrpcRemoteWorker() 56 deleteworkersession_(Method(GrpcWorkerMethod::kDeleteWorkerSession)), in GrpcRemoteWorker() 57 registergraph_(Method(GrpcWorkerMethod::kRegisterGraph)), in GrpcRemoteWorker() 58 deregistergraph_(Method(GrpcWorkerMethod::kDeregisterGraph)), in GrpcRemoteWorker() 59 rungraph_(Method(GrpcWorkerMethod::kRunGraph)), in GrpcRemoteWorker() 60 cleanupgraph_(Method(GrpcWorkerMethod::kCleanupGraph)), in GrpcRemoteWorker() 61 cleanupall_(Method(GrpcWorkerMethod::kCleanupAll)), in GrpcRemoteWorker() 62 recvtensor_(Method(GrpcWorkerMethod::kRecvTensor)), in GrpcRemoteWorker() 63 recvbuf_(Method(GrpcWorkerMethod::kRecvBuf)), in GrpcRemoteWorker() [all …]
|
D | grpc_worker_service_impl.h | 72 enum class GrpcWorkerMethod { enum 91 static_cast<int>(GrpcWorkerMethod::kGetStepSequence) + 1; 93 const char* GrpcWorkerMethodName(GrpcWorkerMethod id);
|
D | grpc_worker_service.cc | 82 static_cast<int>(GrpcWorkerMethod::k##method), \ 90 static_cast<int>(GrpcWorkerMethod::k##method), \ 154 queue_depth_, static_cast<int>(GrpcWorkerMethod::kRecvTensor), in HandleRPCsLoop() 356 static_cast<int>(GrpcWorkerMethod::kRecvTensor), in EnqueueRecvTensorRequestRaw()
|