Function GetInputNodesInfo

serving/processor/framework/graph_optimizer.cc:761–784

}

Status GetInputNodesInfo(std::vector<SrcInfo>* input_info,
                         const Node* node) {
  int edge_count = 0;
  if (IsKvOps(node)) {
    (*input_info).resize(node->num_inputs());
    for (const Edge* edge : node->in_edges()) {
      if (edge->IsControlEdge()) {
        (*input_info).push_back({edge->src(), edge->src_output()});
      } else {
        ++edge_count;
        (*input_info)[edge->dst_input()] = {edge->src(), edge->src_output()};
      }
    }

    if (edge_count != node->num_inputs()) {
      return tensorflow::errors::Internal(
          "Edge count not match, node = ", node->DebugString(),
          ", need ", std::to_string(node->num_inputs()),
          ", got ", std::to_string(edge_count));
    }
  }

  return Status::OK();
}

Status Create1DStringConstOp(const std::string& name,
                             tensorflow::DataType type,