diff options
author | 2018-10-04 12:29:50 -0700 | |
---|---|---|
committer | 2018-10-04 12:36:24 -0700 | |
commit | 900d115135656229e3667025f925eb92687dce18 (patch) | |
tree | 31ee19149765eb8d8d75118fb5d8f3b48c8a3896 /tensorflow/compiler/xla/service/BUILD | |
parent | 5bdd0f7c2807ed413cfc60319f1e75b1e6a4a5b5 (diff) |
[XLA] Move FusionQueue class declaration into separate header
PiperOrigin-RevId: 215783391
Diffstat (limited to 'tensorflow/compiler/xla/service/BUILD')
-rw-r--r-- | tensorflow/compiler/xla/service/BUILD | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/BUILD b/tensorflow/compiler/xla/service/BUILD index f329a27e14..2f8bab0614 100644 --- a/tensorflow/compiler/xla/service/BUILD +++ b/tensorflow/compiler/xla/service/BUILD @@ -1324,10 +1324,19 @@ cc_library( ) cc_library( + name = "fusion_queue", + hdrs = ["fusion_queue.h"], + deps = [ + ":hlo", + ], +) + +cc_library( name = "instruction_fusion", srcs = ["instruction_fusion.cc"], hdrs = ["instruction_fusion.h"], deps = [ + ":fusion_queue", ":hlo", ":hlo_pass", "//tensorflow/compiler/xla:util", |