diff options
author | Justin Lebar <jlebar@google.com> | 2017-09-28 23:07:12 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-28 23:14:07 -0700 |
commit | 9b1b5d85b9ce3c812dc772da1f3f5d09581e5b49 (patch) | |
tree | 798207232f7ae8240d5ee2e7f934669ed1318bef /tensorflow/compiler/xla/tests/fusion_test.cc | |
parent | 75e07e01a41434fdf40eea6291fe7bc47ad74312 (diff) |
[XLA] Make HloComputation::instructions() return a view of HloInstruction*s.
Currently it returns a view of unique_ptr<HloInstruction>s. But the
fact that these are unique_ptrs is an implementation detail, and it's
ugly to leak it everywhere.
PiperOrigin-RevId: 170445375
Diffstat (limited to 'tensorflow/compiler/xla/tests/fusion_test.cc')
-rw-r--r-- | tensorflow/compiler/xla/tests/fusion_test.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/tests/fusion_test.cc b/tensorflow/compiler/xla/tests/fusion_test.cc index 2be409561a..3bf9ccb197 100644 --- a/tensorflow/compiler/xla/tests/fusion_test.cc +++ b/tensorflow/compiler/xla/tests/fusion_test.cc @@ -655,10 +655,10 @@ XLA_TEST_F(FusionTest, SharedConstant) { HloComputation* entry_comp = hlo_module->entry_computation(); // entry computation contains the constant(0) and the fusion - EXPECT_EQ(entry_comp->instructions().size(), 2); + EXPECT_EQ(entry_comp->instruction_count(), 2); // fused instruction contains the constant(2), the parameter, and 4 adds - EXPECT_EQ(entry_comp->root_instruction()->fused_instructions().size(), 6); + EXPECT_EQ(entry_comp->root_instruction()->fused_instruction_count(), 6); LiteralTestUtil::ExpectEqual(*Literal::CreateR1<int32>({8}), *ExecuteAndTransfer(std::move(hlo_module), {})); |