Method AddConsumer

tensorflow/lite/delegates/gpu/common/model.h:298–320

  }

  Status AddConsumer(NodeId consumer, ValueId value) final {
    ValueDef* v;
    RETURN_IF_ERROR(LookupValue(value, &v));
    Value<TensorT>* value_ptr = v->value.get();
    NodeDef* n;
    RETURN_IF_ERROR(LookupNode(consumer, &n));
    Node* node_ptr = n->node.get();

    // check if this value has the same producer already
    if (node_ptr == v->producer) {
      return InvalidArgumentError("Node is a producer of the value");
    }

    // check if this value has the same consumer already
    if (std::find(n->inputs.begin(), n->inputs.end(), value_ptr) !=
        n->inputs.end()) {
      return InvalidArgumentError("Node is already a consumer of the value");
    }

    n->inputs.push_back(value_ptr);
    v->consumers.push_back(node_ptr);
    return OkStatus();
  }

  Status RemoveConsumer(NodeId consumer, ValueId value) final {
    ValueDef* v;