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);