Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dc_api_internal.h113 std::set<tensorflow::string> colocation_constraints; member
Dc_api.cc812 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()
1024 desc->colocation_constraints.clear(); in TF_SetAttrValueProto()
1026 desc->colocation_constraints.insert(location); in TF_SetAttrValueProto()
1044 if (!desc->colocation_constraints.empty()) { in TF_FinishOperationLocked()
1047 std::vector<string>(desc->colocation_constraints.begin(), in TF_FinishOperationLocked()
1048 desc->colocation_constraints.end())); in TF_FinishOperationLocked()
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental_graph.cc234 op_->colocation_constraints.clear(); in SetAttrStringList()
236 op_->colocation_constraints.emplace(static_cast<const char*>(values[i]), in SetAttrStringList()