Searched refs:colocation_constraints (Results 1 – 3 of 3) sorted by relevance
231 op_->colocation_constraints.clear(); in SetAttrStringList()233 op_->colocation_constraints.emplace(static_cast<const char*>(values[i]), in SetAttrStringList()
812 desc->colocation_constraints.emplace( in TF_ColocateWith()826 desc->colocation_constraints.clear(); in TF_SetAttrStringList()828 desc->colocation_constraints.emplace(static_cast<const char*>(values[i]), in TF_SetAttrStringList()1015 desc->colocation_constraints.clear(); in TF_SetAttrValueProto()1017 desc->colocation_constraints.insert(location); in TF_SetAttrValueProto()1035 if (!desc->colocation_constraints.empty()) { in TF_FinishOperationLocked()1038 std::vector<string>(desc->colocation_constraints.begin(), in TF_FinishOperationLocked()1039 desc->colocation_constraints.end())); in TF_FinishOperationLocked()
113 std::set<tensorflow::string> colocation_constraints; member