commit c025ecf14633357a047d98cb1b55f9d24c8a85f3
author A. Unique TensorFlower <gardener@tensorflow.org> Mon Dec 14 13:29:19 2020 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Dec 14 13:33:09 2020 -0800
tree 6fb002550122ec1b285499afd2264c3fcd817fb5
parent df96df98624b3973d45bebd5ed7297469a5addb4

Stop using grpc::internal::ClientAsyncResponseReaderFactory (deprecated)

PiperOrigin-RevId: 347456833
Change-Id: I8943183b8dac5e80fcffc8faeddeec1267a1308c

tensorflow/core/tpu/kernels/tpu_compilation_cache_grpc.cc

diff --git a/tensorflow/core/tpu/kernels/tpu_compilation_cache_grpc.cc b/tensorflow/core/tpu/kernels/tpu_compilation_cache_grpc.cc
index c3aa628..1b21ad8 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::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<ResponseType>(
+          channel_.get(), cq, rpcmethod_get_tpu_program_, context, request);
+  result->StartCall();
+  return result;
 }
 
 ::grpc::ClientAsyncResponseReader<
@@ -76,9 +79,9 @@
 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() {