diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-04 15:51:17 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-04 15:54:13 -0700 |
commit | cf01d118ef0762c0554611bef123bf4559071fbf (patch) | |
tree | c2cf861ad626b3f167a792bfa0df74e8285ca346 /tensorflow/compiler/xla/service/hlo_parser_test.cc | |
parent | 69613d25c3f82652c636c5a1c1b42029dc427979 (diff) |
Add support for kDomain parsing in HLO parser.
PiperOrigin-RevId: 199208527
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 9a18b4f845..84a981675f 100644 --- a/tensorflow/compiler/xla/service/hlo_parser_test.cc +++ b/tensorflow/compiler/xla/service/hlo_parser_test.cc @@ -236,6 +236,17 @@ ENTRY %ShardedTupleCreate.v4 (v1: f32[], v2: f32[3], v3: f32[2,3]) -> (f32[], f3 )" }, +{ +"DomainParsing", +R"(HloModule DomainParsing_module + +ENTRY %DomainParsing (v1: f32[]) -> f32[] { + %v1 = f32[] parameter(0) + ROOT %dom = f32[] domain(f32[] %v1), domain={kind="sharding", entry={maximal device=0}, exit={maximal device=1}} +} + +)" +}, // int32 result = 0; // while (result < 5) { result = result + 1; } { |