1# Description: 2# Utilities that perform useful transformations on graphs 3 4load( 5 "//tensorflow:tensorflow.bzl", 6 "tf_cc_binary", 7 "tf_cuda_library", 8) 9load("//tensorflow:tensorflow.bzl", "get_compatible_with_cloud") 10 11package( 12 default_visibility = ["//visibility:public"], 13 licenses = ["notice"], 14) 15 16tf_cuda_library( 17 name = "optimization_pass_runner_lib", 18 srcs = ["optimization_pass_runner.cc"], 19 hdrs = ["optimization_pass_runner.h"], 20 compatible_with = get_compatible_with_cloud(), 21 deps = [ 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/core:core_cpu", 42 "//tensorflow/core:core_cpu_base", 43 "//tensorflow/core:framework", 44 "//tensorflow/core:framework_internal", 45 "//tensorflow/core:framework_lite", 46 "//tensorflow/core:lib", 47 "//tensorflow/core:protos_all_cc", 48 "//tensorflow/core:tensorflow", 49 "@com_google_absl//absl/strings", 50 ], 51) 52