description_(arg_def.description()),
      arg_type_(arg_type),
      position_(position) {}

ArgSpec ArgSpec::CreateInput(const OpDef::ArgDef& arg_def, int position) {
  if (arg_def.is_ref()) {
    return ArgSpec(arg_def, ArgType::CreateInputRef(arg_def), position);
  } else {
    return ArgSpec(arg_def, ArgType::CreateInput(arg_def), position);
  }
}

ArgSpec ArgSpec::CreateOutput(const OpDef::ArgDef& arg_def, int position) {

//
// This represents the type information with OpDef::ArgDef and any type-related
// context.
class ArgType {
 public:
  ArgType() = default;
  ArgType(const ArgType& other) = default;
  static ArgType CreateInput(const OpDef::ArgDef& arg_def);
  static ArgType CreateInputRef(const OpDef::ArgDef& arg_def);
  static ArgType CreateOutput(const OpDef::ArgDef& arg_def);

  const tensorflow::DataType data_type() const { return data_type_; }

==============================================================================*/
#include "tensorflow/c/experimental/ops/gen/model/arg_type.h"

namespace tensorflow {
namespace generator {

ArgType ArgType::CreateInput(const OpDef::ArgDef& arg_def) {
  return ArgType(arg_def, kInput);
}

ArgType ArgType::CreateInputRef(const OpDef::ArgDef& arg_def) {
  return ArgType(arg_def, kInputRef);
}

//
// Essentially, this represents an OpDef::ArgDef and its context within the Op.
class ArgSpec {
 public:
  ArgSpec() = default;
  ArgSpec(const ArgSpec& other) = default;
  static ArgSpec CreateInput(const OpDef::ArgDef& arg_def, int position);
  static ArgSpec CreateOutput(const OpDef::ArgDef& arg_def, int position);

  const string& name() const { return name_; }

      description_(api_def.description()) {
  absl::flat_hash_set<string> inferred_attrs;
  // Parse the arguments
  for (const OpDef::ArgDef& arg_def : op_def.input_arg()) {
    ArgSpec arg = ArgSpec::CreateInput(arg_def, input_args_.size());
    input_args_.push_back(arg);
    if (!arg_def.type_attr().empty()) {
      inferred_attrs.insert(arg_def.type_attr());
      if (!arg_def.number_attr().empty()) {