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

PiperOrigin-RevId: 347456833
Change-Id: I8943183b8dac5e80fcffc8faeddeec1267a1308c
This commit is contained in:
A. Unique TensorFlower 2020-12-14 13:29:19 -08:00 committed by TensorFlower Gardener
parent df96df9862
commit c025ecf146

View File

@ -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<ResponseType>(
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() {