From 8cd3ca309bc815c7c7a24289519d52364f370d9b Mon Sep 17 00:00:00 2001 From: Gunhan Gulsoy <gunan@google.com> Date: Fri, 20 Sep 2019 23:20:53 -0700 Subject: [PATCH] Move error_codes.proto part 2. Move the usages to protobuf/error_codes.proto PiperOrigin-RevId: 270414415 --- tensorflow/cc/training/coordinator.h | 2 +- tensorflow/cc/training/coordinator_test.cc | 2 +- tensorflow/cc/training/queue_runner.h | 2 +- tensorflow/cc/training/queue_runner_test.cc | 2 +- tensorflow/core/protobuf/master.proto | 2 +- tensorflow/core/protobuf/queue_runner.proto | 4 +++- tensorflow/core/protobuf/worker.proto | 2 +- 7 files changed, 9 insertions(+), 7 deletions(-) diff --git a/tensorflow/cc/training/coordinator.h b/tensorflow/cc/training/coordinator.h index 7168b775251..6d92d05803d 100644 --- a/tensorflow/cc/training/coordinator.h +++ b/tensorflow/cc/training/coordinator.h @@ -22,11 +22,11 @@ limitations under the License. #include <vector> #include "tensorflow/core/framework/cost_graph.pb.h" -#include "tensorflow/core/lib/core/error_codes.pb.h" #include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/platform/macros.h" #include "tensorflow/core/platform/mutex.h" #include "tensorflow/core/protobuf/config.pb.h" +#include "tensorflow/core/protobuf/error_codes.pb.h" namespace tensorflow { diff --git a/tensorflow/cc/training/coordinator_test.cc b/tensorflow/cc/training/coordinator_test.cc index 48874033841..ff4f3f10312 100644 --- a/tensorflow/cc/training/coordinator_test.cc +++ b/tensorflow/cc/training/coordinator_test.cc @@ -17,11 +17,11 @@ limitations under the License. #include "tensorflow/cc/training/queue_runner.h" #include "tensorflow/core/lib/core/blocking_counter.h" -#include "tensorflow/core/lib/core/error_codes.pb.h" #include "tensorflow/core/lib/core/notification.h" #include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/platform/env.h" #include "tensorflow/core/platform/test.h" +#include "tensorflow/core/protobuf/error_codes.pb.h" #include "tensorflow/core/public/session.h" namespace tensorflow { diff --git a/tensorflow/cc/training/queue_runner.h b/tensorflow/cc/training/queue_runner.h index 21189b4b046..d9ecd221493 100644 --- a/tensorflow/cc/training/queue_runner.h +++ b/tensorflow/cc/training/queue_runner.h @@ -23,11 +23,11 @@ limitations under the License. #include "tensorflow/cc/training/coordinator.h" #include "tensorflow/core/lib/core/blocking_counter.h" -#include "tensorflow/core/lib/core/error_codes.pb.h" #include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/lib/core/threadpool.h" #include "tensorflow/core/platform/mutex.h" #include "tensorflow/core/protobuf/config.pb.h" +#include "tensorflow/core/protobuf/error_codes.pb.h" #include "tensorflow/core/protobuf/queue_runner.pb.h" #include "tensorflow/core/public/session.h" diff --git a/tensorflow/cc/training/queue_runner_test.cc b/tensorflow/cc/training/queue_runner_test.cc index 075bc8a55af..b9cdc46ef9d 100644 --- a/tensorflow/cc/training/queue_runner_test.cc +++ b/tensorflow/cc/training/queue_runner_test.cc @@ -25,11 +25,11 @@ limitations under the License. #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" -#include "tensorflow/core/lib/core/error_codes.pb.h" #include "tensorflow/core/lib/core/notification.h" #include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/platform/env.h" #include "tensorflow/core/platform/test.h" +#include "tensorflow/core/protobuf/error_codes.pb.h" #include "tensorflow/core/protobuf/queue_runner.pb.h" #include "tensorflow/core/public/session.h" diff --git a/tensorflow/core/protobuf/master.proto b/tensorflow/core/protobuf/master.proto index 9addf67908c..ecb7057f91c 100644 --- a/tensorflow/core/protobuf/master.proto +++ b/tensorflow/core/protobuf/master.proto @@ -26,8 +26,8 @@ option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobu import "tensorflow/core/framework/device_attributes.proto"; import "tensorflow/core/framework/graph.proto"; import "tensorflow/core/framework/tensor.proto"; -import "tensorflow/core/lib/core/error_codes.proto"; import "tensorflow/core/protobuf/config.proto"; +import "tensorflow/core/protobuf/error_codes.proto"; import "tensorflow/core/protobuf/named_tensor.proto"; //////////////////////////////////////////////////////////////////////////////// diff --git a/tensorflow/core/protobuf/queue_runner.proto b/tensorflow/core/protobuf/queue_runner.proto index f4df649f7d6..c189ff3a570 100644 --- a/tensorflow/core/protobuf/queue_runner.proto +++ b/tensorflow/core/protobuf/queue_runner.proto @@ -1,12 +1,14 @@ syntax = "proto3"; package tensorflow; + option cc_enable_arenas = true; option java_outer_classname = "QueueRunnerProtos"; option java_multiple_files = true; option java_package = "org.tensorflow.framework"; + option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf"; -import "tensorflow/core/lib/core/error_codes.proto"; +import "tensorflow/core/protobuf/error_codes.proto"; // Protocol buffer representing a QueueRunner. message QueueRunnerDef { diff --git a/tensorflow/core/protobuf/worker.proto b/tensorflow/core/protobuf/worker.proto index 88fb76de657..8b2afd7bf36 100644 --- a/tensorflow/core/protobuf/worker.proto +++ b/tensorflow/core/protobuf/worker.proto @@ -31,9 +31,9 @@ import "tensorflow/core/framework/step_stats.proto"; import "tensorflow/core/framework/tensor.proto"; import "tensorflow/core/framework/tensor_shape.proto"; import "tensorflow/core/framework/types.proto"; -import "tensorflow/core/lib/core/error_codes.proto"; import "tensorflow/core/protobuf/config.proto"; import "tensorflow/core/protobuf/debug.proto"; +import "tensorflow/core/protobuf/error_codes.proto"; import "tensorflow/core/protobuf/named_tensor.proto"; import "tensorflow/core/protobuf/tensorflow_server.proto";