}
  return tensorflow::errors::Unimplemented(
      "Only functions with inputs/outputs containing a single tensor or a tuple"
      " of tensors are supported right now.");
}

Status VerifySupportedArgs(TaggedValue args) {
  if (args.type() == TaggedValue::Type::TENSOR) {
    return ::tensorflow::OkStatus();
  }
  if (args.type() == TaggedValue::Type::TUPLE) {
    for (const auto& t : args.tuple()) {