1# Description:
2#   Utilities that perform useful transformations on graphs
3
4package(default_visibility = ["//visibility:public"])
5
6licenses(["notice"])  # Apache 2.0
7
8load(
9    "//tensorflow:tensorflow.bzl",
10    "tf_cc_binary",
11    "tf_cuda_library",
12)
13
14exports_files(["LICENSE"])
15
16tf_cuda_library(
17    name = "optimization_pass_runner_lib",
18    srcs = ["optimization_pass_runner.cc"],
19    hdrs = ["optimization_pass_runner.h"],
20    deps = [
21        "//tensorflow/contrib:contrib_ops_op_lib",
22        "//tensorflow/core:core_cpu",
23        "//tensorflow/core:core_cpu_base",
24        "//tensorflow/core:framework",
25        "//tensorflow/core:framework_internal",
26        "//tensorflow/core:framework_lite",
27        "//tensorflow/core:lib",
28        "//tensorflow/core:protos_all_cc",
29        "//tensorflow/core:tensorflow",
30    ],
31)
32
33tf_cc_binary(
34    name = "gpu_optimization_pass_runner",
35    srcs = ["gpu_optimization_pass_runner_main.cc"],
36    deps = [
37        ":optimization_pass_runner_lib",
38        "//tensorflow/compiler/jit:xla_cpu_jit",
39        "//tensorflow/compiler/jit:xla_gpu_jit",
40        "//tensorflow/compiler/tf2xla:xla_compiler",
41        "//tensorflow/contrib:contrib_ops_op_lib",
42        "//tensorflow/core:core_cpu",
43        "//tensorflow/core:core_cpu_base",
44        "//tensorflow/core:framework",
45        "//tensorflow/core:framework_internal",
46        "//tensorflow/core:framework_lite",
47        "//tensorflow/core:lib",
48        "//tensorflow/core:protos_all_cc",
49        "//tensorflow/core:tensorflow",
50        "@com_google_absl//absl/strings",
51    ],
52)
53