Method BuildEagerOp

tensorflow/lite/delegates/flex/kernel.cc:235–266

Build thew new EagerOperation. In case of error, the returned 'op' is guaranteed to be 'nullptr'.

  // Build thew new EagerOperation. In case of error, the returned 'op' is
  // guaranteed to be 'nullptr'.
  tensorflow::Status BuildEagerOp(tensorflow::EagerContext* eager_context) {
    op_.reset();

    const tensorflow::AttrTypeMap* attr_types;
    bool is_function = false;
    TF_RETURN_WITH_CONTEXT_IF_ERROR(
        tensorflow::AttrTypeMapForOp(name_.c_str(), &attr_types, &is_function),
        " (while processing attributes of '", name_, "')");
    if (is_function) {
      return tensorflow::errors::NotFound(
          "Operation '", name_,
          "' is not registered.  (while processing attributes of '", name_,
          "')");
    }

    op_.reset(new tensorflow::EagerOperation(eager_context, name_.c_str(),
                                             /*is_function=*/false,
                                             attr_types));

    op_->MutableAttrs()->NumInputs(inputs_.Size());
    for (const auto& attr : nodedef_.attr()) {
      op_->MutableAttrs()->Set(attr.first, attr.second);
    }

    // Precalculating a cache key saves about 10% of inference time for very
    // small models.
    tensorflow::Device* device = op_->Device();
    op_->MutableAttrs()->CacheKey(device == nullptr ? "unspecified"
                                                    : device->name());

    return tensorflow::Status::OK();
  }

  void ClearEagerInputs() {
    for (tensorflow::TensorHandle* h : *op_->MutableInputs()) {