diff options
author | 2018-05-31 17:19:25 -0700 | |
---|---|---|
committer | 2018-05-31 17:25:30 -0700 | |
commit | c3b62c38ebd73c98ffa5613865f4c01fa5ff6ae7 (patch) | |
tree | 9b46cb1ea8dab31384b99f5b4ea523c9fc4259b6 /tensorflow/compiler/xla/service/BUILD | |
parent | 30faaee8154575f834050590ebe0bf6ff3f9c176 (diff) |
[XLA] Fix handling of CustomCall's window and dnums.
CustomCall can have a window and convolution-dimension-numbers, so
HloInstruction needs to handle this in Clone() and Identical().
PiperOrigin-RevId: 198805211
Diffstat (limited to 'tensorflow/compiler/xla/service/BUILD')
-rw-r--r-- | tensorflow/compiler/xla/service/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/BUILD b/tensorflow/compiler/xla/service/BUILD index aa416312ad..2b14b63ea8 100644 --- a/tensorflow/compiler/xla/service/BUILD +++ b/tensorflow/compiler/xla/service/BUILD @@ -426,6 +426,7 @@ tf_cc_test( "//tensorflow/compiler/xla:test", "//tensorflow/compiler/xla:test_helpers", "//tensorflow/compiler/xla:util", + "//tensorflow/compiler/xla:window_util", "//tensorflow/compiler/xla/tests:hlo_test_base", "//tensorflow/compiler/xla/tests:xla_internal_test_main", "//tensorflow/compiler/xla/tools/parser:hlo_parser", |