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

ArgType ArgType::CreateOutput(const OpDef::ArgDef& arg_def) {
  return ArgType(arg_def, kOutput);
}

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

// 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_; }