diff options
author | 2018-07-27 10:47:49 -0700 | |
---|---|---|
committer | 2018-07-27 10:53:42 -0700 | |
commit | e888f2cb5ed6f60526a23063bfb30b27f9233980 (patch) | |
tree | 9ae218612b144d9404d3a965e959a2899e625eb1 /tensorflow/compiler/xla/service/hlo_parser_test.cc | |
parent | c72748b673a1b76b1b7b12376a772d04edb16154 (diff) |
[XLA] Parse window and dim_labels that appear on a custom call.
XLA:GPU uses a custom-call with window/dim_labels to represent a call to
cudnn.
PiperOrigin-RevId: 206339219
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_parser_test.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_parser_test.cc | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_parser_test.cc b/tensorflow/compiler/xla/service/hlo_parser_test.cc index 1f0572c576..1c08c51220 100644 --- a/tensorflow/compiler/xla/service/hlo_parser_test.cc +++ b/tensorflow/compiler/xla/service/hlo_parser_test.cc @@ -1016,6 +1016,17 @@ ENTRY Iota { } )" +}, +// custom-call with window and dim_labels +{ +"CustomCallWithWindowAndDimLabels", +R"(HloModule CustomCallWithWindowAndDimLabels + +ENTRY Computation { + ROOT r = f32[100]{0} custom-call(), window={size=2x2}, dim_labels=b01f_01io->b01f, custom_call_target="target" +} + +)" } }); // clang-format on |