for (it = attrs.begin(); it != attrs.end(); it++) {
    if (count == i) {
      return it->first.length();
    }
    count++;
  }
  return -1;
}

void TF_OperationGetAttrName(TF_Operation* oper, int i, char* output,
                             TF_Status* status) {
  auto attrs = oper->node.attrs();
  int count = 0;
  AttrValueMap::const_iterator it;

  ASSERT_EQ(1, TF_OperationGetNumAttrs(oper));
  ASSERT_EQ(1, TF_OperationGetAttrNameLength(oper, 0));

  std::unique_ptr<char[]> value(new char[1]);

  TF_OperationGetAttrName(oper, 0, value.get(), s_);
  EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
  EXPECT_EQ("v", string(static_cast<const char*>(value.get()), 1));
}

TEST_F(CApiAttributesTest, Errors) {