diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/BUILD')
| -rw-r--r-- | tensorflow/compiler/xla/service/BUILD | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/BUILD b/tensorflow/compiler/xla/service/BUILD index 0b85d6febc..cef55f66dc 100644 --- a/tensorflow/compiler/xla/service/BUILD +++ b/tensorflow/compiler/xla/service/BUILD @@ -281,6 +281,38 @@ cc_library( ], ) +cc_library( + name = "call_inliner", + srcs = ["call_inliner.cc"], + hdrs = ["call_inliner.h"], + deps = [ + ":hlo_pass", + "//tensorflow/compiler/xla:statusor", + "//tensorflow/core:lib", + ], +) + +cc_test( + name = "call_inliner_test", + size = "small", + srcs = ["call_inliner_test.cc"], + deps = [ + ":call_inliner", + ":hlo", + ":hlo_matchers", + ":hlo_pass", + "//tensorflow/compiler/xla:literal_util", + "//tensorflow/compiler/xla:shape_util", + "//tensorflow/compiler/xla:test", + "//tensorflow/compiler/xla:types", + "//tensorflow/compiler/xla:util", + "//tensorflow/compiler/xla:xla_data_proto", + "//tensorflow/compiler/xla/tests:hlo_test_base", + "//tensorflow/core:lib", + "//tensorflow/core:test", + ], +) + cc_test( name = "flatten_call_graph_test", size = "small", |