diff options
author | 2018-09-06 00:36:01 -0700 | |
---|---|---|
committer | 2018-09-06 00:42:44 -0700 | |
commit | c200cecbec679cc9dbb219fd06663232f18470ff (patch) | |
tree | 56e2afde205274c233528eec273b37568d25fcb1 /tensorflow/compiler/xla/service/hlo_parser_test.cc | |
parent | 830c8a480a4a65540e60b638cd73b50801408c9b (diff) |
Parse feature_group_count attributes of CustomCall ops.
PiperOrigin-RevId: 211762464
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_parser_test.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_parser_test.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_parser_test.cc b/tensorflow/compiler/xla/service/hlo_parser_test.cc index 0dfc0a4d1c..43e8736532 100644 --- a/tensorflow/compiler/xla/service/hlo_parser_test.cc +++ b/tensorflow/compiler/xla/service/hlo_parser_test.cc @@ -1123,13 +1123,13 @@ ENTRY Iota { )" }, -// custom-call with window and dim_labels +// custom-call with window, dim_labels and feature_group_count { -"CustomCallWithWindowAndDimLabels", -R"(HloModule CustomCallWithWindowAndDimLabels +"CustomCallWithWindowAndDimLabelsAndFeatureGroupCount", +R"(HloModule CustomCallWithWindowAndDimLabelsAndFeatureGroupCount ENTRY Computation { - ROOT r = f32[100]{0} custom-call(), window={size=2x2}, dim_labels=b01f_01io->b01f, custom_call_target="target" + ROOT r = f32[100]{0} custom-call(), window={size=2x2}, dim_labels=b01f_01io->b01f, feature_group_count=2, custom_call_target="target" } )" |