Home
last modified time | relevance | path

Searched refs:GetNodeAttrString (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc115 const string& colo = GetNodeAttrString(n_attrs, kColocationAttrName); in RewriteIntoTempVariable()
264 const string& colo = GetNodeAttrString(n_attrs, kColocationAttrName); in RewriteIntoAddN()
Dparallel_concat_optimizer.cc54 const string& colo = GetNodeAttrString(n_attrs, "_class"); in Run()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc267 const std::string& embedding_attr_string = GetNodeAttrString( in Run()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc243 const string& name = GetNodeAttrString(n->attrs(), kXlaClusterIdAttr); in BuildXlaLaunchOps()
Dmark_for_compilation_pass.cc960 GetNodeAttrString(node->attrs(), kXlaInternalScopeAttr); in GetXlaScope()
966 const string& scope = GetNodeAttrString(node->attrs(), kXlaScopeAttr); in GetXlaScope()
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.h313 const std::string& GetNodeAttrString(const AttrSlice& attrs,
Dnode_def_util.cc332 const string& GetNodeAttrString(const AttrSlice& attrs, StringPiece attr_name) { in GetNodeAttrString() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc805 const string& value = GetNodeAttrString(n->attrs(), attr_name); in CheckStringAttr()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc953 const string& send_device = GetNodeAttrString(*ndef, "send_device"); in SetIncarnation()