diff options
author | HyoukJoong Lee <hyouklee@google.com> | 2018-10-05 07:46:22 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-05 07:50:27 -0700 |
commit | 92c8a77ba480bf4aeddea412cc1d2988f6ad81cd (patch) | |
tree | 79d2501bb790c05f1542cf3deaf9fa098b6557af /tensorflow/compiler/xla/service | |
parent | 123de2797a4348c963b597096762085bfa09eab1 (diff) |
Use absl::Span for HloModuleGroupMetadata
PiperOrigin-RevId: 215905026
Diffstat (limited to 'tensorflow/compiler/xla/service')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_module_group_metadata.cc | 2 | ||||
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_module_group_metadata.h | 8 |
2 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc b/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc index 83352ef91b..b4aac4c807 100644 --- a/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc +++ b/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc @@ -59,7 +59,7 @@ string HloModuleGroupMetadata::TrackedInstruction::ToString() const { } /* static */ StatusOr<std::unique_ptr<HloModuleGroupMetadata>> -HloModuleGroupMetadata::Build(const std::vector<HloModule*>& modules) { +HloModuleGroupMetadata::Build(absl::Span<HloModule* const> modules) { auto metadata = absl::make_unique<HloModuleGroupMetadata>(modules); TF_RETURN_IF_ERROR(metadata->Build()); return std::move(metadata); diff --git a/tensorflow/compiler/xla/service/hlo_module_group_metadata.h b/tensorflow/compiler/xla/service/hlo_module_group_metadata.h index 0311b73207..928df0f5a7 100644 --- a/tensorflow/compiler/xla/service/hlo_module_group_metadata.h +++ b/tensorflow/compiler/xla/service/hlo_module_group_metadata.h @@ -102,14 +102,14 @@ class HloModuleGroupMetadata { HloInstruction* recv_done = nullptr; }; - explicit HloModuleGroupMetadata(const std::vector<HloModule*>& modules) - : modules_(modules) {} + explicit HloModuleGroupMetadata(absl::Span<HloModule* const> modules) + : modules_(modules.begin(), modules.end()) {} ~HloModuleGroupMetadata() = default; // Build and return the metadata for the given modules. static StatusOr<std::unique_ptr<HloModuleGroupMetadata>> Build( - const std::vector<HloModule*>& modules); + absl::Span<HloModule* const> modules); // Returns true if the instruction is one of the 4 channel instructions (Send, // Recv, SendDone, RecvDone). @@ -274,7 +274,7 @@ class HloModuleGroupMetadata { int64 max_channel_id_ = -1; // The modules that this metadata was built from. - const std::vector<HloModule*>& modules_; + const std::vector<HloModule*> modules_; absl::flat_hash_map<HloModule*, std::unique_ptr<TuplePointsToAnalysis>> points_to_analyses_; |