Searched refs:kColocationAttrName (Results 1 – 10 of 10) 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 | placer.cc | 39 const StringPiece kColocationAttrNameStringPiece(kColocationAttrName);
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_constructor.cc | 533 const auto iter = node_def.attr().find(kColocationAttrName); in ValidateColocationConstraints()
765 if (node_def->attr().find(kColocationAttrName) != node_def->attr().end()) { in AddPrefixToNodeDef()
767 node_def->mutable_attr()->at(kColocationAttrName).mutable_list(); in AddPrefixToNodeDef()
808 GetNodeAttr(node->attrs(), kColocationAttrName, &coloc_values); in UpdateUniquifiedColocationNames()
822 node->AddAttr(kColocationAttrName, coloc_values); in UpdateUniquifiedColocationNames()
|
| D | graph_constructor_test.cc | 148 Status s = GetNodeAttr(n->attrs(), kColocationAttrName, &value); in ColocationGroup()
1988 results.return_nodes[1]->attrs().Find(kColocationAttrName); in TEST_F()
2003 class_attr = results.return_nodes[0]->attrs().Find(kColocationAttrName); in TEST_F()
2019 class_attr = results.return_nodes[0]->attrs().Find(kColocationAttrName); in TEST_F()
2023 class_attr = results.return_nodes[1]->attrs().Find(kColocationAttrName); in TEST_F()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | scope.cc | 218 if (GetNodeAttr(attrs, kColocationAttrName, &node_constraints).ok()) { in GetColocationConstraints()
291 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 | 39 const char* const kColocationAttrName = "_class"; variable
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 1202 TF_SetAttrStringList(desc, tensorflow::kColocationAttrName, list_ptrs.get(), in SetViaStringList()
1214 TF_SetAttrValueProto(desc, tensorflow::kColocationAttrName, bytes.data(), in SetViaProto()
1222 TF_OperationGetAttrMetadata(op, tensorflow::kColocationAttrName, s_); in VerifyCollocation()
1236 TF_OperationGetAttrStringList(op, tensorflow::kColocationAttrName, in VerifyCollocation()
|
| D | c_api.cc | 1144 if (strcmp(attr_name, tensorflow::kColocationAttrName) == 0) { in TF_SetAttrStringList()
1328 if (strcmp(attr_name, tensorflow::kColocationAttrName) == 0) { in TF_SetAttrValueProto()
1333 tensorflow::kColocationAttrName, "\" attribute"); in TF_SetAttrValueProto()
1358 tensorflow::kColocationAttrName, in TF_FinishOperationLocked()
|