// when moving out of experimental.
  static SignatureDefFunction* wrap(TF_SignatureDefFunction* p) {
    return reinterpret_cast<SignatureDefFunction*>(p);
  }
  static TF_SignatureDefFunction* unwrap(SignatureDefFunction* p) {
    return reinterpret_cast<TF_SignatureDefFunction*>(p);
  }
};

inline const SignatureDefFunctionMetadata*
SignatureDefFunction::GetFunctionMetadata() {
  return SignatureDefFunctionMetadata::wrap(

#ifdef __cplusplus
extern "C" {
#endif  // __cplusplus

// An opaque type that corresponds to a SignatureDefFunction loaded from a
// SavedModel.
typedef struct TF_SignatureDefFunctionMetadata TF_SignatureDefFunctionMetadata;

// Retrieves the arguments of the SignatureDefFunction. The caller is not
// responsible for freeing the returned pointer.
TF_CAPI_EXPORT extern const TF_SignatureDefParamList*

// the input/output names of the SignatureDef:
// https://github.com/tensorflow/tensorflow/blob/9bcefa44cd335c1db4a703a13da09f29ae1bbdb2/tensorflow/core/protobuf/meta_graph.proto#L318-L321
class SignatureDefFunction {
 public:
  virtual ~SignatureDefFunction() = default;

  // Creates a "Call" Op used to execute the function.
  virtual Status MakeCallOp(absl::Span<AbstractTensorHandle* const> inputs,

  // Retrieve a SignatureDefFunction from a SavedModel, using the key of the
  // SignatureDef map:
  // https://github.com/tensorflow/tensorflow/blob/69b08900b1e991d84bce31f3b404f5ed768f339f/tensorflow/core/protobuf/meta_graph.proto#L89
  virtual Status GetSignatureDefFunction(const std::string& signature_def_key,
                                         SignatureDefFunction** function) = 0;

}

inline SignatureDefFunction* SavedModelAPI::GetSignatureDefFunction(
    const std::string& function_path, Status* status) {
  TF_SignatureDefFunction* function = TF_GetSavedModelSignatureDefFunction(
      saved_model_.get(), function_path.c_str(), status->GetTFStatus());
  if (!status->ok()) {
    return nullptr;
  }
  return SignatureDefFunction::wrap(function);
}

}  // namespace cc

#include "tensorflow/core/framework/function.pb.h"
#include "tensorflow/core/protobuf/saved_object_graph.pb.h"

namespace tensorflow {

// This is the TF eager runtime implementation of SignatureDefFunction (separate
// from the TFRT implementation). The user-facing API of SignatureDefFunctions
// and their semantic differences from ConcreteFunction are described here:

#include "tensorflow/c/experimental/saved_model/core/signature_def_function.h"

typedef struct TF_SignatureDefFunction TF_SignatureDefFunction;

namespace tensorflow {

DEFINE_CONVERSION_FUNCTIONS(tensorflow::SignatureDefFunction,
                            TF_SignatureDefFunction)

}  // namespace tensorflow

#include "tensorflow/core/platform/status.h"
#include "tensorflow/core/protobuf/struct.pb.h"

namespace tensorflow {

// SignatureDefParam represents a named Tensor input or output to a
// SignatureDefFunction.
class SignatureDefParam {
 public:
  SignatureDefParam(std::string name, TensorSpec spec);

  const std::string& name() const;

  const TensorSpec& spec() const;

 private:
  std::string name_;

// and output's names, dtypes, and shape.
class SignatureDefFunctionMetadata final {
  // TODO(bmzhao): Add getters here as necessary.
 private:
  friend class SignatureDefFunction;
  static SignatureDefFunctionMetadata* wrap(
      TF_SignatureDefFunctionMetadata* p) {
    return reinterpret_cast<SignatureDefFunctionMetadata*>(p);
  }
  static TF_SignatureDefFunctionMetadata* unwrap(

#include "tensorflow/core/framework/function.pb.h"
#include "tensorflow/core/protobuf/saved_object_graph.pb.h"

namespace tensorflow {

// FunctionBuilder wraps the state needed for building a SignatureDefFunction.
// This is mainly used in PartiallyRevivedObjects, which wraps partially
// constructed Function and Resource objects.
struct TFSignatureDefFunctionRevivalState {