absl::Status SetAttrShape(AbstractOperation*, const char* attr_name,
                          const int64_t* dims, const int num_dims,
                          ForwardOperation*);
absl::Status SetAttrFunction(AbstractOperation*, const char* attr_name,
                             const AbstractOperation* value, ForwardOperation*);
absl::Status SetAttrFunctionName(AbstractOperation*, const char* attr_name,

                                    const int num_dims) = 0;
  virtual absl::Status SetAttrShape(const char* attr_name,
                                    const PartialTensorShape shape);
  virtual absl::Status SetAttrFunction(const char* attr_name,
                                       const AbstractOperation* value) = 0;
  virtual absl::Status SetAttrFunctionName(const char* attr_name,
                                           const char* value,

    }
    op_->node_builder.Attr(attr_name, shape);
    return absl::OkStatus();
  }
  absl::Status SetAttrFunction(const char* attr_name,
                               const AbstractOperation* value) override {
    return tensorflow::errors::Unimplemented(
        "SetAttrFunction has not been implemented yet.");
  }
  absl::Status SetAttrFunctionName(const char* attr_name, const char* value,

  return op_->SetAttrShape(attr_name, dims, num_dims);
}
absl::Status SetAttrFunction(AbstractOperation* op_, const char* attr_name,
                             const AbstractOperation* value,
                             ForwardOperation* forward_op_) {
  return absl::UnimplementedError(
      "SetAttrFunction has not been implemented yet.");
}

}

void TFE_OpSetAttrFunction(TFE_Op* op, const char* attr_name,
                           const TFE_Op* value) {
  auto s = tensorflow::unwrap(op)->SetAttrFunction(
      attr_name, tensorflow::unwrap(const_cast<TFE_Op*>(value)));
  if (!s.ok()) {
    LOG(WARNING) << "Unable to set attribute: " << attr_name;
  }
}