Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dgrpc_tpu_driver.cc775 auto grpc_handle = static_cast<GrpcBufferHandle*>(handle.get()); in Deallocate() local
776 req->mutable_dealloc()->set_handle(grpc_handle->id().AsInt()); in Deallocate()
868 auto grpc_handle = static_cast<const GrpcCompiledProgramHandle*>(handle); in LoadProgram() local
869 if (grpc_handle->id().client_id != driver_->client_id()) { in LoadProgram()
876 req->mutable_load()->set_compiled_program_handle(grpc_handle->id().AsInt()); in LoadProgram()
916 auto* grpc_handle = static_cast<GrpcBufferHandle*>(input); in ExecuteProgram() local
917 if (grpc_handle->id().client_id != driver_->client_id()) { in ExecuteProgram()
922 req->mutable_execute()->add_input_handle(grpc_handle->id().AsInt()); in ExecuteProgram()
926 auto* grpc_handle = static_cast<GrpcBufferHandle*>(output); in ExecuteProgram() local
927 if (grpc_handle->id().client_id != driver_->client_id()) { in ExecuteProgram()