Change HloBuffer, HloBufferSet and HloValueSet to hold pointers rather than ids.

This makes it easier to implement logic like returning the size of an HloBuffer,
which requires knowing the underlying HloValues.

No functional changes; only a change of representation.

PiperOrigin-RevId: 161737042

1 files changed, 3 insertions, 3 deletions

diff --git a/tensorflow/compiler/xla/service/hlo_dataflow_analysis_test.cc b/tensorflow/compiler/xla/service/hlo_dataflow_analysis_test.cc
index af7e22f9e1..2b685e355f 100644
--- a/tensorflow/compiler/xla/service/hlo_dataflow_analysis_test.cc
+++ b/tensorflow/compiler/xla/service/hlo_dataflow_analysis_test.cc
@@ -56,9 +56,9 @@ class HloDataflowAnalysisTest : public HloTestBase,
                                     const ShapeIndex& index = {}) {
     CHECK(analysis_ != nullptr);
     std::vector<HloValue> values;
-    for (HloValue::Id value_id :
-         analysis_->GetValueSet(instruction, index).value_ids()) {
-      values.push_back(analysis_->GetValue(value_id));
+    for (const HloValue* value :
+         analysis_->GetValueSet(instruction, index).values()) {
+      values.push_back(*value);
     }
     return values;
   }