Function RemoveFollowingNode

tensorflow/lite/delegates/gpu/common/model.h:504–521

// that are produced by to_keep. to_keep inherits all to_remove inputs.
template <typename TensorT>
Status RemoveFollowingNode(Graph<TensorT>* graph, const Node* to_remove,
                           const Node* to_keep) {
  // Make sure all inputs to to_remove are produced by to_keep.
  for (auto input : graph->FindInputs(to_remove->id)) {
    Node* producer = graph->FindProducer(input->id);
    if (producer->id != to_keep->id) {
      return InvalidArgumentError("To_remove node has other inputs");
    }
  }

  for (auto input : graph->FindInputs(to_remove->id)) {
    RETURN_IF_ERROR(graph->DeleteValue(input->id));
  }
  for (auto output : graph->FindOutputs(to_remove->id)) {
    RETURN_IF_ERROR(graph->SetProducer(to_keep->id, output->id));
  }
  return graph->DeleteNode(to_remove->id);
}

// Removes to_remove node.
// Requires that node has one input and one output;