1licenses(["restricted"]) 2 3package(default_visibility = ["//visibility:public"]) 4 5load( 6 "//tensorflow/core:platform/default/build_config_root.bzl", 7 "if_static", 8) 9 10cc_library( 11 name = "interpreter_transfer_manager", 12 srcs = ["interpreter_transfer_manager.cc"], 13 hdrs = ["interpreter_transfer_manager.h"], 14 deps = [ 15 "//tensorflow/compiler/xla:util", 16 "//tensorflow/compiler/xla:xla_data_proto", 17 "//tensorflow/compiler/xla/service:generic_transfer_manager", 18 "//tensorflow/compiler/xla/service:transfer_manager", 19 "//tensorflow/compiler/xla/service/interpreter:platform_id", 20 "//tensorflow/core:lib", 21 "//tensorflow/core:stream_executor_no_cuda", 22 ], 23 alwayslink = True, # Contains per-platform transfer manager registration 24) 25 26cc_library( 27 name = "compiler", 28 srcs = ["compiler.cc"], 29 hdrs = ["compiler.h"], 30 deps = [ 31 ":executable", 32 ":platform_id", 33 "//tensorflow/compiler/xla:status", 34 "//tensorflow/compiler/xla:status_macros", 35 "//tensorflow/compiler/xla:statusor", 36 "//tensorflow/compiler/xla:util", 37 "//tensorflow/compiler/xla:xla_data_proto", 38 "//tensorflow/compiler/xla/service:algebraic_simplifier", 39 "//tensorflow/compiler/xla/service:compiler", 40 "//tensorflow/compiler/xla/service:computation_placer", 41 "//tensorflow/compiler/xla/service:executable", 42 "//tensorflow/compiler/xla/service:flatten_call_graph", 43 "//tensorflow/compiler/xla/service:hlo", 44 "//tensorflow/compiler/xla/service:hlo_constant_folding", 45 "//tensorflow/compiler/xla/service:hlo_cost_analysis", 46 "//tensorflow/compiler/xla/service:hlo_cse", 47 "//tensorflow/compiler/xla/service:hlo_dce", 48 "//tensorflow/compiler/xla/service:hlo_module_config", 49 "//tensorflow/compiler/xla/service:hlo_pass", 50 "//tensorflow/compiler/xla/service:hlo_pass_pipeline", 51 "//tensorflow/compiler/xla/service:hlo_subcomputation_unification", 52 "//tensorflow/compiler/xla/service:inliner", 53 "//tensorflow/compiler/xla/service:layout_assignment", 54 "//tensorflow/compiler/xla/service:reshape_mover", 55 "//tensorflow/compiler/xla/service:while_loop_simplifier", 56 "//tensorflow/core:lib", 57 "//tensorflow/stream_executor", 58 ], 59 alwayslink = True, # Contains compiler registration 60) 61 62cc_library( 63 name = "platform_id", 64 srcs = ["platform_id.cc"], 65 hdrs = ["platform_id.h"], 66 deps = [ 67 "@nsync//:nsync_headers", 68 "//tensorflow/core:stream_executor_headers_lib", 69 ] + if_static( 70 ["@protobuf_archive//:protobuf"], 71 ["@protobuf_archive//:protobuf_headers"], 72 ), 73) 74 75cc_library( 76 name = "executable", 77 srcs = ["executable.cc"], 78 hdrs = ["executable.h"], 79 deps = [ 80 ":executor", 81 "//tensorflow/compiler/xla:literal_util", 82 "//tensorflow/compiler/xla:shape_util", 83 "//tensorflow/compiler/xla:status_macros", 84 "//tensorflow/compiler/xla:statusor", 85 "//tensorflow/compiler/xla:types", 86 "//tensorflow/compiler/xla:util", 87 "//tensorflow/compiler/xla:xla_data_proto", 88 "//tensorflow/compiler/xla/service:executable", 89 "//tensorflow/compiler/xla/service:hlo", 90 "//tensorflow/compiler/xla/service:hlo_cost_analysis", 91 "//tensorflow/compiler/xla/service:hlo_evaluator", 92 "//tensorflow/compiler/xla/service:hlo_execution_profile", 93 "//tensorflow/compiler/xla/service:hlo_module_config", 94 "//tensorflow/compiler/xla/service:shaped_buffer", 95 "//tensorflow/compiler/xla/service:transfer_manager", 96 "//tensorflow/core:lib", 97 "//tensorflow/core:stream_executor_no_cuda", 98 ], 99) 100 101cc_library( 102 name = "platform", 103 srcs = ["platform.cc"], 104 hdrs = ["platform.h"], 105 deps = [ 106 ":executor", 107 ":platform_id", 108 "//tensorflow/core:stream_executor_headers_lib", 109 ], 110 alwayslink = True, # Registers itself with the MultiPlatformManager. 111) 112 113cc_library( 114 name = "executor", 115 srcs = ["executor.cc"], 116 hdrs = ["executor.h"], 117 deps = [ 118 "//tensorflow/compiler/xla:shape_util", 119 "//tensorflow/compiler/xla:status_macros", 120 "//tensorflow/compiler/xla:xla_data_proto", 121 "//tensorflow/core:lib", 122 "//tensorflow/core:stream_executor_headers_lib", 123 "//tensorflow/core:stream_executor_no_cuda", 124 ], 125) 126 127filegroup( 128 name = "all_files", 129 srcs = glob( 130 ["**/*"], 131 exclude = [ 132 "**/METADATA", 133 "**/OWNERS", 134 ], 135 ), 136) 137