Function MarkComputeGraph

tensorflow/core/graph/graph_util.cc:26–48

namespace {
void MarkComputeGraph(const Graph* g, std::vector<bool>& is_var_relate) {
  // Get the starting node of the coloring algorithm
  std::queue<const Node*> q;
  for (Node* n : g->op_nodes()) {
    if (n->IsVariable() || n->IsKvVarHandle()
        || n->IsControlFlow()
        || n->type_string() == "VarHandleOp") {
      q.emplace(n);
    }
  }

  // mark compute graph node
  while (!q.empty()) {
    const Node* node = q.front();
    q.pop();
    is_var_relate[node->id()] = true;
    for (auto e : node->out_edges()) {
      if (!is_var_relate[e->dst()->id()]) {
        q.emplace(e->dst());
      }
    }
  }
}

void DealWithNode(Graph* g, Node* n,
      const std::vector<bool>& is_var_relate,