extern "C" {

extern const char* TF_SignatureDefParamName(const TF_SignatureDefParam* param) {
  return tensorflow::unwrap(param)->name().c_str();
}

extern const TF_TensorSpec* TF_SignatureDefParamTensorSpec(
    const TF_SignatureDefParam* param) {
  return tensorflow::wrap(&tensorflow::unwrap(param)->spec());
}

// Returns the TensorSpec associated with the given parameter. The caller is
// not reponsible for freeing the returned TF_TensorSpec*.
TF_CAPI_EXPORT extern const TF_TensorSpec* TF_SignatureDefParamTensorSpec(
    const TF_SignatureDefParam* param);

#ifdef __cplusplus
}  // end extern "C"
#endif  // __cplusplus

  EXPECT_EQ(TF_SignatureDefParamListSize(args), 2);
  const TF_SignatureDefParam* param_a = TF_SignatureDefParamListGet(args, 0);
  const TF_TensorSpec* tensor_spec_a = TF_SignatureDefParamTensorSpec(param_a);
  const TF_Shape* shape_a = TF_TensorSpecShape(tensor_spec_a);

  // Input "a" is a scalar, float32 tensor
  EXPECT_EQ("a", std::string(TF_SignatureDefParamName(param_a)));