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/call_graph.h | |
| parent | 44acd839c57494860666c799afd24360f1df3bed (diff) | |
[XLA] Migrate from gtl::FlatMap to absl::flat_hash_map
PiperOrigin-RevId: 215272497
Diffstat (limited to 'tensorflow/compiler/xla/service/call_graph.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/call_graph.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/compiler/xla/service/call_graph.h b/tensorflow/compiler/xla/service/call_graph.h index 3af2ab5edf..0c2e9b99db 100644 --- a/tensorflow/compiler/xla/service/call_graph.h +++ b/tensorflow/compiler/xla/service/call_graph.h @@ -20,10 +20,10 @@ limitations under the License. #include <ostream> +#include "absl/container/flat_hash_map.h" #include "tensorflow/compiler/xla/service/hlo_computation.h" #include "tensorflow/compiler/xla/service/hlo_instruction.h" #include "tensorflow/compiler/xla/service/hlo_module.h" -#include "tensorflow/core/lib/gtl/flatmap.h" #include "tensorflow/core/lib/gtl/flatset.h" namespace xla { @@ -157,7 +157,7 @@ class CallGraphNode { // The map from instruction to index in callsites_ for looking up the callsite // (if any) associated with a particular instruction in this computation. - tensorflow::gtl::FlatMap<const HloInstruction*, int64> callsite_instructions_; + absl::flat_hash_map<const HloInstruction*, int64> callsite_instructions_; // The call sites in other computations which call this computation. std::vector<CallSite> caller_callsites_; @@ -267,7 +267,7 @@ class CallGraph { // Map from HLO computation to the index of the corresponding call graph node // in nodes_. - tensorflow::gtl::FlatMap<const HloComputation*, int64> node_indices_; + absl::flat_hash_map<const HloComputation*, int64> node_indices_; }; } // namespace xla |