#include "tensorflow/c/experimental/saved_model/internal/signature_def_param_list_type.h"

extern "C" {

extern const TF_SignatureDefParamList* TF_SignatureDefFunctionMetadataArgs(
    const TF_SignatureDefFunctionMetadata* list) {
  return tensorflow::wrap(&tensorflow::unwrap(list)->arguments());
}

extern const TF_SignatureDefParamList* TF_SignatureDefFunctionMetadataReturns(

// Retrieves the arguments of the SignatureDefFunction. The caller is not
// responsible for freeing the returned pointer.
TF_CAPI_EXPORT extern const TF_SignatureDefParamList*
TF_SignatureDefFunctionMetadataArgs(
    const TF_SignatureDefFunctionMetadata* list);

// Retrieves the returns of the SignatureDefFunction. The caller is not
// responsible for freeing the returned pointer.

  TF_SignatureDefFunctionMetadata* metadata =
      TF_SignatureDefFunctionGetMetadata(serving_default);

  const TF_SignatureDefParamList* args =
      TF_SignatureDefFunctionMetadataArgs(metadata);
  const TF_SignatureDefParamList* returns =
      TF_SignatureDefFunctionMetadataReturns(metadata);

  EXPECT_EQ(TF_SignatureDefParamListSize(args), 2);