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;
}