Home
last modified time | relevance | path

Searched refs:ToGrpcStatus (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
Dservice.cc87 return ToGrpcStatus(xla::InvalidArgument("Invalid protocol version %d", in Connect()
94 return ToGrpcStatus(tensorflow::errors::Aborted( in Connect()
100 return ToGrpcStatus(status); in Connect()
119 return ToGrpcStatus(tensorflow::errors::DeadlineExceeded( in Connect()
136 return ToGrpcStatus( in Connect()
163 return ToGrpcStatus(status); in Shutdown()
168 return ToGrpcStatus(service_status_); in Shutdown()
170 return ToGrpcStatus(xla::FailedPrecondition( in Shutdown()
176 return ToGrpcStatus(status); in Shutdown()
187 return ToGrpcStatus(tensorflow::errors::DeadlineExceeded( in Shutdown()
[all …]
Dutil.h33 inline ::grpc::Status ToGrpcStatus(const Status& s) { in ToGrpcStatus() function
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_master_service.cc157 call->SendResponse(ToGrpcStatus(status)); in CreateSessionHandler()
168 call->SendResponse(ToGrpcStatus(status)); in ExtendSessionHandler()
178 call->SendResponse(ToGrpcStatus(status)); in PartialRunSetupHandler()
207 call->SendResponse(ToGrpcStatus(Status::OK())); in RunStepHandler()
209 call->SendResponse(ToGrpcStatus(status)); in RunStepHandler()
220 call->SendResponse(ToGrpcStatus(status)); in CloseSessionHandler()
230 call->SendResponse(ToGrpcStatus(status)); in ListDevicesHandler()
239 call->SendResponse(ToGrpcStatus(status)); in ResetHandler()
249 call->SendResponse(ToGrpcStatus(status)); in MakeCallableHandler()
269 call->SendResponse(ToGrpcStatus(status)); in RunCallableHandler()
[all …]
Dgrpc_worker_service.cc199 call->SendResponse(ToGrpcStatus(s)); \
227 call->SendResponse(ToGrpcStatus(s)); in GetStepSequenceHandler()
260 call->SendResponse(ToGrpcStatus(s)); in RunGraphHandler()
280 call->SendResponse(ToGrpcStatus(s)); in RecvTensorHandlerRaw()
297 call->SendResponse(ToGrpcStatus(s)); in RecvBufHandler()
316 call->SendResponse(ToGrpcStatus(s)); in CompleteGroupHandler()
335 call->SendResponse(ToGrpcStatus(s)); in CompleteInstanceHandler()
Dgrpc_util.h97 inline ::grpc::Status ToGrpcStatus(const ::tensorflow::Status& s) { in ToGrpcStatus() function
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/
Dgrpc_eager_service_impl.h59 ToGrpcStatus(local_impl_.method(&call->request, &call->response))); \
78 call->SendResponse(ToGrpcStatus(local_impl_.Enqueue( in EnqueueHandler()
99 call->SendResponse(ToGrpcStatus(s)); in RunComponentFunctionHandler()
142 call->Finish(ToGrpcStatus(status)); in StreamingEnqueueHandler()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compilation_cache_service.cc110 return call->SendResponse(ToGrpcStatus(s)); in GetTpuProgram()
135 return call->SendResponse(ToGrpcStatus(buffer_slices.status())); in GetTpuProgram()
/external/tensorflow/tensorflow/core/data/service/
Dgrpc_worker_impl.cc43 return ToGrpcStatus(impl_.method(request, response)); \
Dgrpc_dispatcher_impl.cc41 return ToGrpcStatus(impl_.method(request, response)); \
/external/tensorflow/tensorflow/compiler/xla/rpc/
Dgrpc_service.cc32 return tensorflow::ToGrpcStatus(s); in DelegateRPC()