Home
last modified time | relevance | path

Searched refs:colocation_constraints (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/grappler/
Dcontroller.py58 self.colocation_constraints = item.GetColocationGroups()
/external/tensorflow/tensorflow/c/
Dc_api_internal.h121 std::set<tensorflow::string> colocation_constraints; member
Dc_api.cc1234 desc->colocation_constraints.emplace( in TF_ColocateWith()
1248 desc->colocation_constraints.clear(); in TF_SetAttrStringList()
1250 desc->colocation_constraints.emplace(static_cast<const char*>(values[i]), in TF_SetAttrStringList()
1446 desc->colocation_constraints.clear(); in TF_SetAttrValueProto()
1448 desc->colocation_constraints.insert(location); in TF_SetAttrValueProto()
1466 if (!desc->colocation_constraints.empty()) { in TF_FinishOperationLocked()
1469 std::vector<string>(desc->colocation_constraints.begin(), in TF_FinishOperationLocked()
1470 desc->colocation_constraints.end())); in TF_FinishOperationLocked()