Searched refs:colocation_graph_ (Results 1 – 1 of 1) sorted by relevance
76 : colocation_graph_(colocation_graph), next_group_id_(0) {} in ColocationGraphToIOColocationGroups()81 int root_id = colocation_graph_->FindAndUpdateRoot(nodes[i]->id()); in AssignGroups()100 const Member& member = colocation_graph_->members()[it.first]; in FillGroups()107 ColocationGraph* colocation_graph_; member in tensorflow::ColocationGraphToIOColocationGroups