From c025ecf14633357a047d98cb1b55f9d24c8a85f3 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 14 Dec 2020 13:29:19 -0800 Subject: [PATCH] Stop using grpc::internal::ClientAsyncResponseReaderFactory (deprecated) PiperOrigin-RevId: 347456833 Change-Id: I8943183b8dac5e80fcffc8faeddeec1267a1308c --- .../tpu/kernels/tpu_compilation_cache_grpc.cc | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/tensorflow/core/tpu/kernels/tpu_compilation_cache_grpc.cc b/tensorflow/core/tpu/kernels/tpu_compilation_cache_grpc.cc index c3aa62805c0..1b21ad8bc70 100644 --- a/tensorflow/core/tpu/kernels/tpu_compilation_cache_grpc.cc +++ b/tensorflow/core/tpu/kernels/tpu_compilation_cache_grpc.cc @@ -66,9 +66,12 @@ grpc::TpuCompilationCacheService::Stub::Stub( grpc::TpuCompilationCacheService::Stub::AsyncGetTpuProgramRaw( ::grpc::ClientContext* context, const RequestType& request, ::grpc::CompletionQueue* cq) { - return ::grpc::internal::ClientAsyncResponseReaderFactory< - ResponseType>::Create(channel_.get(), cq, rpcmethod_get_tpu_program_, - context, request, true); + ::grpc::ClientAsyncResponseReader< + grpc::TpuCompilationCacheService::ResponseType>* result = + ::grpc::internal::ClientAsyncResponseReaderHelper::Create( + channel_.get(), cq, rpcmethod_get_tpu_program_, context, request); + result->StartCall(); + return result; } ::grpc::ClientAsyncResponseReader< @@ -76,9 +79,9 @@ grpc::TpuCompilationCacheService::Stub::AsyncGetTpuProgramRaw( grpc::TpuCompilationCacheService::Stub::PrepareAsyncGetTpuProgramRaw( ::grpc::ClientContext* context, const RequestType& request, ::grpc::CompletionQueue* cq) { - return ::grpc::internal::ClientAsyncResponseReaderFactory< - ResponseType>::Create(channel_.get(), cq, rpcmethod_get_tpu_program_, - context, request, false); + return ::grpc::internal::ClientAsyncResponseReaderHelper::Create< + ResponseType>(channel_.get(), cq, rpcmethod_get_tpu_program_, context, + request); } grpc::TpuCompilationCacheService::Service::Service() {