Function ReplaceNode

serving/processor/framework/graph_optimizer.cc:666–707

namespace {

Status ReplaceNode(
    const std::string& op_name,
    Node* node, Graph* graph,
    std::vector<SrcInfo>& input_info,
    std::unordered_map<std::string, const AttrValue*>& attr_info) {

  // Create KvLookup/KvImport op here and remove the node
  NodeDef new_node_def;
  new_node_def.set_name(node->name());
  new_node_def.set_op(op_name);

  // Set attrs
  for (auto attr : attr_info) {
    (*new_node_def.mutable_attr())[attr.first] = *(attr.second);
  }

  Status status;
  Node *new_node = graph->AddNode(new_node_def, &status);
  if (!status.ok()) return status;

  // Add input egdes, from slot 0 -> n
  for (size_t i = 0; i < input_info.size(); ++i) {
    int idx = i;
    if (input_info[i].src_slot == Graph::kControlSlot) {
      idx = Graph::kControlSlot;
    }
    graph->AddEdge(input_info[i].src_node,
                   input_info[i].src_slot,
                   new_node, idx);
  }

  // Add output edges
  for (const Edge* edge : node->out_edges()) {
    graph->AddEdge(new_node, edge->src_output(),
                   edge->dst(), edge->dst_input());
  }

  // remove current node
  graph->RemoveNode(node);

  return Status::OK();
}

Status ReplaceNode(
    const std::string& op_name,