diff options
author | 2017-01-23 21:27:24 -0800 | |
---|---|---|
committer | 2017-01-23 21:42:46 -0800 | |
commit | 863bab34202f650282dfe00aaa082a4796fdd839 (patch) | |
tree | 72e2ca9c0baaaddc8c19dbb3662cda31f62e59b5 /tensorflow/compiler/xla/shape_util_test.cc | |
parent | a544628dc9fa3607c2a398ac72f5e859fd151e94 (diff) |
Improvements to HLO text format printing.
Change: 145374835
Diffstat (limited to 'tensorflow/compiler/xla/shape_util_test.cc')
-rw-r--r-- | tensorflow/compiler/xla/shape_util_test.cc | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/tensorflow/compiler/xla/shape_util_test.cc b/tensorflow/compiler/xla/shape_util_test.cc index fb2f8fb284..9e6b243611 100644 --- a/tensorflow/compiler/xla/shape_util_test.cc +++ b/tensorflow/compiler/xla/shape_util_test.cc @@ -338,13 +338,12 @@ TEST(ShapeUtilTest, HumanString) { EXPECT_EQ("opaque[]", ShapeUtil::HumanStringWithLayout(opaque)); EXPECT_EQ("f32[]", ShapeUtil::HumanStringWithLayout(scalar)); - EXPECT_EQ("u32[1,2] {1,0}", ShapeUtil::HumanStringWithLayout(matrix)); - EXPECT_EQ("s32[3,4] {0,1}", ShapeUtil::HumanStringWithLayout(matrix2)); - EXPECT_EQ("(opaque[], f32[], u32[1,2] {1,0}, s32[3,4] {0,1})", + EXPECT_EQ("u32[1,2]{1,0}", ShapeUtil::HumanStringWithLayout(matrix)); + EXPECT_EQ("s32[3,4]{0,1}", ShapeUtil::HumanStringWithLayout(matrix2)); + EXPECT_EQ("(opaque[], f32[], u32[1,2]{1,0}, s32[3,4]{0,1})", ShapeUtil::HumanStringWithLayout(tuple)); - EXPECT_EQ( - "((opaque[], f32[], u32[1,2] {1,0}, s32[3,4] {0,1}), u32[1,2] {1,0})", - ShapeUtil::HumanStringWithLayout(nested_tuple)); + EXPECT_EQ("((opaque[], f32[], u32[1,2]{1,0}, s32[3,4]{0,1}), u32[1,2]{1,0})", + ShapeUtil::HumanStringWithLayout(nested_tuple)); ProgramShape prog = ShapeUtil::MakeProgramShape( {opaque, scalar, matrix, matrix2, tuple, nested_tuple}, nested_tuple); |