From f15c3b0d02ca8556d37b8879e8e1529a7b0276fe Mon Sep 17 00:00:00 2001
From: Yujing Zhang <yujingzhang@google.com>
Date: Fri, 13 Sep 2019 13:43:57 -0700
Subject: [PATCH] Change OpNameToAttrTypeMap from std::unordered_map<string,
 const AttrTypeMap*> to tensorflow::gtl::FlatMap<string, const AttrTypeMap*>

PiperOrigin-RevId: 268972276
---
 tensorflow/core/common_runtime/eager/attr_builder.cc | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tensorflow/core/common_runtime/eager/attr_builder.cc b/tensorflow/core/common_runtime/eager/attr_builder.cc
index eeef9b33921..55eecbcbce2 100644
--- a/tensorflow/core/common_runtime/eager/attr_builder.cc
+++ b/tensorflow/core/common_runtime/eager/attr_builder.cc
@@ -32,8 +32,9 @@ namespace {
 
 mutex g_op_name_to_attr_type_map_lock(LINKER_INITIALIZED);
 
-std::unordered_map<string, const AttrTypeMap*>* OpNameToAttrTypeMap() {
-  static auto* const m = new std::unordered_map<string, const AttrTypeMap*>;
+tensorflow::gtl::FlatMap<string, const AttrTypeMap*>* OpNameToAttrTypeMap() {
+  static auto* const m =
+      new tensorflow::gtl::FlatMap<string, const AttrTypeMap*>;
   return m;
 }