From 21fda280bc72bdbc4386c7b0d2ad4b97264921ad Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 18 Jul 2022 20:09:44 -0700 Subject: [PATCH] Use pb.h for topology.proto. This change uses pb.h instead of proto.h because OSS proto codegen doesn't support proto.h. The OSS conversion converts proto.h to pb.h in some directories but model_servers/ and examples/ are explicitly skipped. PiperOrigin-RevId: 461780646 --- tensorflow_serving/model_servers/main.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_serving/model_servers/main.cc b/tensorflow_serving/model_servers/main.cc index d19ec45bdca..d6664c12c89 100644 --- a/tensorflow_serving/model_servers/main.cc +++ b/tensorflow_serving/model_servers/main.cc @@ -55,7 +55,7 @@ limitations under the License. #include "tensorflow_serving/model_servers/version.h" #if defined(LIBTPU_ON_GCE) || defined(PLATFORM_CLOUD_TPU) -#include "tensorflow/core/protobuf/tpu/topology.proto.h" +#include "tensorflow/core/protobuf/tpu/topology.pb.h" #include "tensorflow/core/tpu/tpu_global_init.h" void InitializeTPU(tensorflow::serving::main::Server::Options& server_options) {