Home
last modified time | relevance | path

Searched refs:kClassAttr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dcolocation.cc73 constexpr char kClassAttr[] = "_class"; in ReassignColocation() local
80 auto iter = node.attr().find(kClassAttr); in ReassignColocation()
107 node->mutable_attr()->erase(kClassAttr); in ReassignColocation()
108 node->mutable_attr()->insert({kClassAttr, new_value}); in ReassignColocation()
115 node->mutable_attr()->erase(kClassAttr); in ReassignColocation()
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.cc901 constexpr char kClassAttr[] = "_class"; in ADD_ATTR() local
902 auto class_attr = node_def->mutable_attr()->find(kClassAttr); in ADD_ATTR()
907 node_def->mutable_attr()->erase(kClassAttr); in ADD_ATTR()
908 node_def->mutable_attr()->insert({kClassAttr, new_value}); in ADD_ATTR()