Searched refs:kColocationAttrName (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | accumulate_n_optimizer.cc | 84 if (GetNodeAttr(n_attrs, kColocationAttrName, &colo).ok()) { in rewriteNode()
85 node_builder.Attr(kColocationAttrName, colo); in rewriteNode()
|
| D | process_function_library_runtime.cc | 285 static const StringPiece kColocationAttrNameStringPiece(kColocationAttrName); in GetColocationGroup()
351 node->AddAttr(kColocationAttrName, colo_slice); in PinArgsAndRets()
|
| D | colocation_graph.cc | 46 const StringPiece kColocationAttrNameStringPiece(kColocationAttrName);
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_constructor.cc | 561 const auto iter = node_def.attr().find(kColocationAttrName); in ValidateColocationConstraints()
811 if (node_def->attr().find(kColocationAttrName) != node_def->attr().end()) { in AddPrefixToNodeDef()
813 node_def->mutable_attr()->at(kColocationAttrName).mutable_list(); in AddPrefixToNodeDef()
854 GetNodeAttr(node->attrs(), kColocationAttrName, &coloc_values); in UpdateUniquifiedColocationNames()
868 node->AddAttr(kColocationAttrName, coloc_values); in UpdateUniquifiedColocationNames()
|
| D | graph_constructor_test.cc | 148 Status s = GetNodeAttr(n->attrs(), kColocationAttrName, &value); in ColocationGroup()
1997 results.return_nodes[1]->attrs().Find(kColocationAttrName); in TEST_F()
2012 class_attr = results.return_nodes[0]->attrs().Find(kColocationAttrName); in TEST_F()
2028 class_attr = results.return_nodes[0]->attrs().Find(kColocationAttrName); in TEST_F()
2032 class_attr = results.return_nodes[1]->attrs().Find(kColocationAttrName); in TEST_F()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | scope.cc | 275 if (GetNodeAttr(attrs, kColocationAttrName, &node_constraints).ok()) { in GetColocationConstraints()
347 builder->Attr(kColocationAttrName, constraints); in UpdateBuilder()
|
| D | cc_ops_test.cc | 37 TF_EXPECT_OK(GetNodeAttr(tensor.op().node()->attrs(), kColocationAttrName, in GetColocationConstraints()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.h | 43 extern const char* const kColocationAttrName;
|
| D | node_def_util.cc | 40 const char* const kColocationAttrName = "_class"; variable
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 1223 TF_SetAttrStringList(desc, tensorflow::kColocationAttrName, list_ptrs.get(), in SetViaStringList()
1235 TF_SetAttrValueProto(desc, tensorflow::kColocationAttrName, bytes.data(), in SetViaProto()
1243 TF_OperationGetAttrMetadata(op, tensorflow::kColocationAttrName, s_); in VerifyCollocation()
1257 TF_OperationGetAttrStringList(op, tensorflow::kColocationAttrName, in VerifyCollocation()
|
| D | c_api.cc | 1247 if (strcmp(attr_name, tensorflow::kColocationAttrName) == 0) { in TF_SetAttrStringList()
1438 if (strcmp(attr_name, tensorflow::kColocationAttrName) == 0) { in TF_SetAttrValueProto()
1443 tensorflow::kColocationAttrName, "\" attribute"); in TF_SetAttrValueProto()
1468 tensorflow::kColocationAttrName, in TF_FinishOperationLocked()
|