| /external/tensorflow/tensorflow/core/common_runtime/ |
| accumulate_n_optimizer.cc | 84 if (GetNodeAttr(n_attrs, kColocationAttrName, &colo).ok()) {
85 node_builder.Attr(kColocationAttrName, colo);
|
| placer.cc | 39 const StringPiece kColocationAttrNameStringPiece(kColocationAttrName);
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| graph_constructor.cc | 533 const auto iter = node_def.attr().find(kColocationAttrName);
765 if (node_def->attr().find(kColocationAttrName) != node_def->attr().end()) {
767 node_def->mutable_attr()->at(kColocationAttrName).mutable_list();
808 GetNodeAttr(node->attrs(), kColocationAttrName, &coloc_values);
822 node->AddAttr(kColocationAttrName, coloc_values);
[all...] |
| graph_constructor_test.cc | 148 Status s = GetNodeAttr(n->attrs(), kColocationAttrName, &value);
[all...] |
| /external/tensorflow/tensorflow/cc/framework/ |
| scope.cc | 218 if (GetNodeAttr(attrs, kColocationAttrName, &node_constraints).ok()) {
291 builder->Attr(kColocationAttrName, constraints);
|
| cc_ops_test.cc | 37 TF_EXPECT_OK(GetNodeAttr(tensor.op().node()->attrs(), kColocationAttrName,
|
| /external/tensorflow/tensorflow/core/framework/ |
| node_def_util.h | 43 extern const char* const kColocationAttrName;
|
| node_def_util.cc | 39 const char* const kColocationAttrName = "_class";
|
| /external/tensorflow/tensorflow/c/ |
| c_api.cc | [all...] |
| c_api_test.cc | [all...] |
