experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change OpNameToAttrTypeMap from std::unordered_map<string, const AttrTypeMap*> to tensorflow::gtl::FlatMap<string, const AttrTypeMap*>

Browse Source 
PiperOrigin-RevId: 268972276
This commit is contained in:
Yujing Zhang 2019-09-13 13:43:57 -07:00 committed by TensorFlower Gardener
parent d24c1171ee
commit f15c3b0d02
1 changed files with 3 additions and 2 deletions

5
tensorflow/core/common_runtime/eager/attr_builder.cc
View File

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