Skip to content

Commit

Permalink
Use pb.h for topology.proto.
Browse files Browse the repository at this point in the history
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
  • Loading branch information
tensorflower-gardener authored and tensorflow-copybara committed Jul 19, 2022
1 parent e1e799f commit 21fda28
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tensorflow_serving/model_servers/main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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) {
Expand Down

0 comments on commit 21fda28

Please sign in to comment.