8000 Automated Code Change · linux-on-ibm-z/tensorflow@7a51c9f · GitHub
[go: up one dir, main page]
Skip to content

Commit 7a51c9f

Browse files
tensorflower-gardenertensorflower-gardener
committed
Automated Code Change
PiperOrigin-RevId: 730731111
1 parent 023d8cc commit 7a51c9f

16 files changed

+40
-5
lines changed

‎tensorflow/compiler/mlir/tf2xla/internal/BUILD

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@ tf_cc_test(
5454
"//tensorflow/core:lib",
5555
"//tensorflow/core/lib/monitoring:cell_reader",
5656
"@com_google_absl//absl/status",
57+
"@com_google_absl//absl/status:statusor",
5758
"@com_google_googletest//:gtest_main",
5859
"@local_xla//xla/hlo/builder:xla_computation",
5960
"@local_xla//xla/service:hlo_proto_cc",
@@ -67,6 +68,7 @@ cc_library(
6768
hdrs = ["mlir_pass_instrumentation.h"],
6869
deps = [
6970
"//tensorflow/core/platform:logging",
71+
"@com_google_absl//absl/log",
7072
"@llvm-project//mlir:Pass",
7173
],
7274
)
@@ -78,6 +80,8 @@ tf_cc_test(
7880
":mlir_pass_instrumentation",
7981
"//tensorflow/compiler/mlir/tf2xla/api/v1:compile_mlir_util_no_tf_dialect_passes",
8082
"//tensorflow/core:test",
83+
"@com_google_absl//absl/strings",
84+
"@com_google_absl//absl/strings:string_view",
8185
"@com_google_googletest//:gtest_main",
8286
],
8387
)
@@ -158,6 +162,7 @@ cc_library(
158162
"//tensorflow/core/tpu/kernels:tpu_compile_op_support",
159163
"@com_google_absl//absl/log",
160164
"@com_google_absl//absl/status",
165+
"@com_google_absl//absl/status:statusor",
161166
"@llvm-project//llvm:Support",
162167
"@llvm-project//mlir:DataLayoutInterfaces",
163168
"@llvm-project//mlir:IR",
@@ -189,6 +194,7 @@ tf_cc_test(
189194
"//tensorflow/core/protobuf:for_core_protos_cc",
190195
"//tensorflow/core/protobuf/tpu:compile_metadata_proto_cc",
191196
"//tensorflow/core/tpu/kernels:tpu_compile_op_support",
197+
"@com_google_absl//absl/status:statusor",
192198
"@com_google_googletest//:gtest",
193199
"@llvm-project//mlir:IR",
194200
"@llvm-project//mlir:Pass",
@@ -255,6 +261,7 @@ tf_cc_test(
255261
"//tensorflow/core:lib",
256262
"//tensorflow/core:test",
257263
"//tensorflow/core/platform:resource_loader",
264+
"@com_google_absl//absl/status",
258265
"@com_google_absl//absl/strings",
259266
"@com_google_googletest//:gtest_main",
260267
"@llvm-project//mlir:FuncDialect",
@@ -277,6 +284,7 @@ cc_library(
277284
"//tensorflow/compiler/tf2xla:tf2xla_defs",
278285
"//tensorflow/core:core_cpu_base",
279286
"//tensorflow/core:framework",
287+
"//tensorflow/core:protos_all_cc",
280288
"//tensorflow/core/common_runtime:function_body",
281289
"@com_google_absl//absl/log",
282290
"@com_google_absl//absl/status",
@@ -307,9 +315,11 @@ tf_cc_test(
307315
"//tensorflow/cc:tpu_ops",
308316
"//tensorflow/compiler/mlir/tensorflow",
309317
"//tensorflow/compiler/tf2xla:tf2xla_defs",
318+
"//tensorflow/core/protobuf:for_core_protos_cc",
310319
"//tensorflow/core:core_cpu_base",
311320
"//tensorflow/core:framework",
312321
"//tensorflow/core:portable_gif_internal",
322+
"//tensorflow/core:protos_all_cc",
313323
"//tensorflow/core:test_main",
314324
"//tensorflow/core:testlib",
315325
"//tensorflow/core/platform:enable_tf2_utils",
@@ -381,8 +391,10 @@ cc_library(
381391
"//tensorflow/core:framework",
382392
"//tensorflow/core:framework_types_hdr",
383393
"//tensorflow/core:lib",
394+
"//tensorflow/core:protos_all_cc",
384395
"//tensorflow/core/common_runtime:function_body",
385396
"//tensorflow/core/platform:enable_tf2_utils",
397+
"//tensorflow/core/protobuf:for_core_protos_cc",
386398
"@com_google_absl//absl/log",
387399
"@com_google_absl//absl/status",
388400
"@com_google_absl//absl/strings:string_view",

‎tensorflow/compiler/mlir/tf2xla/internal/compilation_timer_test.cc

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

1616
#include "tensorflow/compiler/mlir/tf2xla/internal/compilation_timer.h"
1717

18+
#include <cstdint>
19+
1820
#include <gmock/gmock.h>
1921
#include <gtest/gtest.h>
2022
#include "absl/time/clock.h"

‎tensorflow/compiler/mlir/tf2xla/internal/graph_to_tf_executor_util.cc

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,13 @@ limitations under the License.
2626
#include "tensorflow/core/common_runtime/function_body.h"
2727
#include "tensorflow/core/common_runtime/function_def_utils.h"
2828
#include "tensorflow/core/framework/function.h"
29+
#include "tensorflow/core/framework/node_def.pb.h"
2930
#include "tensorflow/core/framework/node_def_util.h"
3031
#include "tensorflow/core/framework/types.h"
32+
#include "tensorflow/core/framework/types.pb.h"
3133
#include "tensorflow/core/graph/graph.h"
3234
#include "tensorflow/core/platform/enable_tf2_utils.h"
35+
#include "tensorflow/core/protobuf/config.pb.h"
3336
#include "tensorflow/core/util/device_name_utils.h"
3437

3538
namespace tensorflow {

‎tensorflow/compiler/mlir/tf2xla/internal/graph_to_tf_executor_util.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ limitations under the License.
1919
#include <optional>
2020

2121
#include "tensorflow/core/graph/graph.h"
22+
#include "tensorflow/core/protobuf/config.pb.h"
2223

2324
namespace tensorflow {
2425

‎tensorflow/compiler/mlir/tf2xla/internal/graph_to_tf_executor_util_test.cc

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,8 @@ limitations under the License.
3333
#include "xla/tsl/lib/core/status_test_util.h"
3434
#include "xla/tsl/platform/status.h"
3535
#include "tensorflow/core/framework/function.h"
36+
#include "tensorflow/core/framework/function.pb.h"
37+
#include "tensorflow/core/framework/node_def.pb.h"
3638
#include "tensorflow/core/framework/node_def_builder.h"
3739
#include "tensorflow/core/framework/node_def_util.h"
3840
#include "tensorflow/core/framework/op.h"
@@ -43,6 +45,7 @@ limitations under the License.
4345
#include "tensorflow/core/graph/graph.h"
4446
#include "tensorflow/core/platform/enable_tf2_utils.h"
4547
#include "tensorflow/core/platform/types.h"
48+
#include "tensorflow/core/protobuf/config.pb.h"
4649

4750
namespace tensorflow {
4851

‎tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.cc

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ limitations under the License.
1616
#include "tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.h"
1717

1818
#include <memory>
19-
#include <string>
2019
#include <vector>
2120

2221
#include "absl/log/log.h"

‎tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,12 @@ limitations under the License.
1919
#include <string>
2020
#include <vector>
2121

22+
#include "absl/status/status.h"
2223
#include "llvm/ADT/StringRef.h"
2324
#include "mlir/Pass/Pass.h" // from @llvm-project
2425
#include "tensorflow/compiler/tf2xla/xla_helpers.h"
2526
#include "xla/tsl/platform/statusor.h"
27+
#include "tensorflow/core/protobuf/tpu/compile_metadata.pb.h"
2628
#include "tensorflow/core/tpu/kernels/tpu_compile_op_support.h"
2729

2830
namespace tensorflow {

‎tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo.cc

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ limitations under the License.
1616
#include "tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo.h"
1717

1818
#include <memory>
19-
#include <string>
2019
#include <vector>
2120

2221
#include "absl/log/log.h"

‎tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,13 @@ limitations under the License.
1616
#ifndef TENSORFLOW_COMPILER_MLIR_TF2XLA_INTERNAL_LEGALIZE_TF_TO_HLO_H_
1717
#define TENSORFLOW_COMPILER_MLIR_TF2XLA_INTERNAL_LEGALIZE_TF_TO_HLO_H_
1818

19+
#include "absl/status/statusor.h"
1920
#include "llvm/ADT/StringRef.h"
2021
#include "mlir/Pass/Pass.h" // from @llvm-project
2122
#include "tensorflow/compiler/tf2xla/xla_helpers.h"
2223
#include "xla/client/compile_only_client.h"
2324
#include "xla/tsl/platform/statusor.h"
25+
#include "tensorflow/core/protobuf/tpu/compile_metadata.pb.h"
2426
#include "tensorflow/core/tpu/kernels/tpu_compile_op_support.h"
2527

2628
namespace tensorflow {

‎tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo_test.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ limitations under the License.
2121

2222
#include <gmock/gmock.h>
2323
#include <gtest/gtest.h>
24+
#include "absl/status/statusor.h"
2425
#include "mlir/IR/BuiltinOps.h" // from @llvm-project
2526
#include "mlir/IR/DialectRegistry.h" // from @llvm-project
2627
#include "mlir/IR/MLIRContext.h" // from @llvm-project

0 commit comments

Comments
 (0)
0