diff options
| author |  Benjamin Kramer <kramerb@google.com> | 2018-10-01 13:43:49 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-01 13:53:07 -0700 |
| commit | 3039a4694e22674b502257ae34b0a5b614a631f3 (patch) | |
| tree | 423fdfa7a2e7dd2740af97accfe848bc97b335d0 /tensorflow/compiler/xla/service/layout_assignment.h | |
| parent | 44acd839c57494860666c799afd24360f1df3bed (diff) | |
[XLA] Migrate from gtl::FlatMap to absl::flat_hash_map
PiperOrigin-RevId: 215272497
Diffstat (limited to 'tensorflow/compiler/xla/service/layout_assignment.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/layout_assignment.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/compiler/xla/service/layout_assignment.h b/tensorflow/compiler/xla/service/layout_assignment.h index e29c199c42..1591256fad 100644 --- a/tensorflow/compiler/xla/service/layout_assignment.h +++ b/tensorflow/compiler/xla/service/layout_assignment.h @@ -25,6 +25,7 @@ limitations under the License. #include <utility> #include <vector> +#include "absl/container/flat_hash_map.h" #include "tensorflow/compiler/xla/service/computation_layout.h" #include "tensorflow/compiler/xla/service/hlo_computation.h" #include "tensorflow/compiler/xla/service/hlo_instruction.h" @@ -38,7 +39,6 @@ limitations under the License. #include "tensorflow/compiler/xla/types.h" #include "tensorflow/compiler/xla/xla_data.pb.h" #include "tensorflow/core/lib/core/status.h" -#include "tensorflow/core/lib/gtl/flatmap.h" #include "tensorflow/core/lib/gtl/flatset.h" #include "tensorflow/core/platform/types.h" @@ -228,8 +228,8 @@ class LayoutConstraints { // Array-shaped buffers which have not yet been constrained. std::set<LogicalBuffer::Id> unconstrained_buffer_ids_; - mutable tensorflow::gtl::FlatMap<const HloInstruction*, - std::unique_ptr<PointsToSet::BufferSet>> + mutable absl::flat_hash_map<const HloInstruction*, + std::unique_ptr<PointsToSet::BufferSet>> buffer_sets_cache_; HloComputation* computation_; |