Skip to content

Commit c22174d

Browse files
tensorflower-gardenertensorflower-gardener
committed
Automated Code Change
PiperOrigin-RevId: 652164801
1 parent 1c1cdf2 commit c22174d

File tree

7 files changed

+54
-13
lines changed

7 files changed

+54
-13
lines changed

‎tensorflow/cc/BUILD

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -911,10 +911,13 @@ cc_library(
911911
deps = [
912912
":coordinator",
913913
"//tensorflow/core:core_cpu",
914+
"//tensorflow/core:framework",
914915
"//tensorflow/core:lib",
915916
"//tensorflow/core:lib_internal",
916917
"//tensorflow/core:protos_all_cc",
917-
"//tensorflow/core/kernels:ops_util",
918+
"@com_google_absl//absl/log",
919+
"@com_google_absl//absl/status",
920+
"@local_tsl//tsl/protobuf:error_codes_proto_impl_cc",
918921
],
919922
)
920923

@@ -924,6 +927,7 @@ tf_cc_test(
924927
deps = [
925928
"coordinator",
926929
":cc_ops",
930+
":ops",
927931
":queue_runner",
928932
":scope",
929933
"//tensorflow/core:core_cpu",
@@ -934,7 +938,7 @@ tf_cc_test(
934938
"//tensorflow/core:tensorflow",
935939
"//tensorflow/core:test",
936940
"//tensorflow/core:test_main",
937-
"//tensorflow/core:testlib",
941+
"@local_tsl//tsl/protobuf:error_codes_proto_impl_cc",
938942
],
939943
)
940944

@@ -946,6 +950,8 @@ cc_library(
946950
"//tensorflow/core:lib",
947951
"//tensorflow/core:lib_internal",
948952
"//tensorflow/core:protos_all_cc",
953+
"@com_google_absl//absl/status",
954+
"@local_tsl//tsl/protobuf:error_codes_proto_impl_cc",
949955
],
950956
)
951957

@@ -955,17 +961,14 @@ tf_cc_test(
955961
deps = [
956962
":cc_ops",
957963
":coordinator",
958-
":queue_runner",
959-
":scope",
960-
"//tensorflow/core:core_cpu",
961-
"//tensorflow/core:framework",
962964
"//tensorflow/core:lib",
963965
"//tensorflow/core:lib_internal",
964966
"//tensorflow/core:protos_all_cc",
965967
"//tensorflow/core:tensorflow",
966968
"//tensorflow/core:test",
967969
"//tensorflow/core:test_main",
968-
"//tensorflow/core:testlib",
970+
"@com_google_absl//absl/status",
971+
"@local_tsl//tsl/protobuf:error_codes_proto_impl_cc",
969972
],
970973
)
971974

‎tensorflow/cc/training/coordinator.cc

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,12 @@ limitations under the License.
1515

1616
#include "tensorflow/cc/training/coordinator.h"
1717

18+
#include "absl/status/status.h"
19+
#include "tensorflow/core/framework/cost_graph.pb.h"
20+
#include "tensorflow/core/platform/mutex.h"
21+
#include "tensorflow/core/platform/status.h"
22+
#include "tsl/protobuf/error_codes.pb.h"
23+
1824
namespace tensorflow {
1925

2026
Coordinator::Coordinator() : Coordinator(std::vector<error::Code>()) {}

‎tensorflow/cc/training/coordinator.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,15 @@ limitations under the License.
2121
#include <unordered_set>
2222
#include <vector>
2323

24+
#include "absl/status/status.h"
2425
#include "tensorflow/core/framework/cost_graph.pb.h"
2526
#include "tensorflow/core/lib/core/status.h"
2627
#include "tensorflow/core/platform/macros.h"
2728
#include "tensorflow/core/platform/mutex.h"
2829
#include "tensorflow/core/protobuf/config.pb.h"
2930
#include "tensorflow/core/protobuf/error_codes.pb.h"
31+
#include "tsl/platform/thread_annotations.h"
32+
#include "tsl/protobuf/error_codes.pb.h"
3033

3134
namespace tensorflow {
3235

‎tensorflow/cc/training/coordinator_test.cc

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,14 +15,16 @@ limitations under the License.
1515

1616
#include "tensorflow/cc/training/coordinator.h"
1717

18-
#include "tensorflow/cc/training/queue_runner.h"
18+
#include "absl/status/status.h"
1919
#include "tensorflow/core/lib/core/notification.h"
20-
#include "tensorflow/core/lib/core/status_test_util.h"
2120
#include "tensorflow/core/platform/blocking_counter.h"
2221
#include "tensorflow/core/platform/env.h"
22+
#include "tensorflow/core/platform/status.h"
2323
#include "tensorflow/core/platform/test.h"
24+
#include "tensorflow/core/platform/threadpool.h"
2425
#include "tensorflow/core/protobuf/error_codes.pb.h"
25-
#include "tensorflow/core/public/session.h"
26+
#include "tsl/lib/core/status_test_util.h"
27+
#include "tsl/protobuf/error_codes.pb.h"
2628

2729
namespace tensorflow {
2830
namespace {

‎tensorflow/cc/training/queue_runner.cc

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,22 @@ limitations under the License.
1414
==============================================================================*/
1515

1616
#include "tensorflow/cc/training/queue_runner.h"
17-
#include "tensorflow/core/kernels/ops_util.h"
17+
18+
#include "absl/log/log.h"
19+
#include "absl/status/status.h"
20+
#include "tensorflow/cc/training/coordinator.h"
21+
#include "tensorflow/core/framework/cost_graph.pb.h"
22+
#include "tensorflow/core/framework/ops_util.h"
23+
#include "tensorflow/core/platform/blocking_counter.h"
1824
#include "tensorflow/core/platform/env.h"
25+
#include "tensorflow/core/platform/mutex.h"
26+
#include "tensorflow/core/platform/status.h"
27+
#include "tensorflow/core/platform/threadpool.h"
28+
#include "tensorflow/core/platform/types.h"
29+
#include "tensorflow/core/protobuf/config.pb.h"
30+
#include "tensorflow/core/protobuf/queue_runner.pb.h"
31+
#include "tensorflow/core/public/session.h"
32+
#include "tsl/protobuf/error_codes.pb.h"
1933

2034
namespace tensorflow {
2135

‎tensorflow/cc/training/queue_runner.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,12 @@ limitations under the License.
2626
#include "tensorflow/core/lib/core/threadpool.h"
2727
#include "tensorflow/core/platform/blocking_counter.h"
2828
#include "tensorflow/core/platform/mutex.h"
29+
#include "tensorflow/core/platform/types.h"
2930
#include "tensorflow/core/protobuf/config.pb.h"
3031
#include "tensorflow/core/protobuf/error_codes.pb.h"
3132
#include "tensorflow/core/protobuf/queue_runner.pb.h"
3233
#include "tensorflow/core/public/session.h"
34+
#include "tsl/platform/thread_annotations.h"
3335

3436
namespace tensorflow {
3537

‎tensorflow/cc/training/queue_runner_test.cc

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,20 +18,31 @@ limitations under the License.
1818
#include <string>
1919
#include <vector>
2020

21+
#include "tensorflow/cc/framework/ops.h"
2122
#include "tensorflow/cc/framework/scope.h"
22-
#include "tensorflow/cc/ops/standard_ops.h"
23+
#include "tensorflow/cc/ops/const_op.h"
24+
#include "tensorflow/cc/ops/data_flow_ops.h"
25+
#include "tensorflow/cc/ops/math_ops.h"
26+
#include "tensorflow/cc/ops/random_ops.h"
27+
#include "tensorflow/cc/ops/state_ops.h"
2328
#include "tensorflow/cc/training/coordinator.h"
29+
#include "tensorflow/core/framework/cost_graph.pb.h"
2430
#include "tensorflow/core/framework/graph.pb.h"
2531
#include "tensorflow/core/framework/tensor.h"
2632
#include "tensorflow/core/framework/tensor_shape.h"
2733
#include "tensorflow/core/framework/types.pb.h"
2834
#include "tensorflow/core/lib/core/notification.h"
29-
#include "tensorflow/core/lib/core/status_test_util.h"
3035
#include "tensorflow/core/platform/env.h"
36+
#include "tensorflow/core/platform/status.h"
3137
#include "tensorflow/core/platform/test.h"
38+
#include "tensorflow/core/protobuf/config.pb.h"
3239
#include "tensorflow/core/protobuf/error_codes.pb.h"
3340
#include "tensorflow/core/protobuf/queue_runner.pb.h"
3441
#include "tensorflow/core/public/session.h"
42+
#include "tensorflow/core/public/session_options.h"
43+
#include "tsl/lib/core/status_test_util.h"
44+
#include "tsl/platform/status.h"
45+
#include "tsl/protobuf/error_codes.pb.h"
3546

3647
namespace tensorflow {
3748
namespace {

0 commit comments

Comments
 (0)