diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-10-24 20:37:06 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-24 20:40:46 -0700 |
commit | 7bd701a29e958be1f836eabc498a67d742c35cdf (patch) | |
tree | d541206da1182d508cd457c1f7a21eb039a5e057 /tensorflow/compiler/xla/service/hlo_instruction_test.cc | |
parent | 557b0b27edff763c165ad59d10d49da8bccbec4f (diff) |
Support more instructions in Hlo parser.
- while, tuple, send/recv, get-tuple-element, call.
- "device="
Also,
- Change HloModule::ToString to print computations in post order, so that a computation is defined before it's used.
- Add % before computation name when it's used.
PiperOrigin-RevId: 173350323
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_instruction_test.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_instruction_test.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_instruction_test.cc b/tensorflow/compiler/xla/service/hlo_instruction_test.cc index cdafc05d8c..9affecae60 100644 --- a/tensorflow/compiler/xla/service/hlo_instruction_test.cc +++ b/tensorflow/compiler/xla/service/hlo_instruction_test.cc @@ -1203,13 +1203,13 @@ TEST_F(HloInstructionTest, Stringification) { EXPECT_EQ(fusion->ToString(false, false), "%fusion = f32[5,20]{1,0} fusion:kTransposeDot(f32[5,10]{1,0} %x, " - "f32[20,10]{1,0} %y), calls=fused_computation"); + "f32[20,10]{1,0} %y), calls=%fused_computation"); HloInstruction* loop = builder.AddInstruction( HloInstruction::CreateWhile(sout, computation, computation, x)); EXPECT_EQ(loop->ToString(false, false), "%while = f32[5,20]{1,0} while(f32[5,10]{1,0} %x), " - "condition=TransposeDot, body=TransposeDot"); + "condition=%TransposeDot, body=%TransposeDot"); } } // namespace |