ASSERT_EQ(TF_OK, TF_GetCode(s.get())) << TF_Message(s.get());
  ASSERT_NE(*func, nullptr);
}

REGISTER_OP("CustomOp")
    .Output("output: float32")
    .Attr("index: int")
    .SetShapeFn(tensorflow::shape_inference::UnknownShape);

void NodeWithPlaceholderAttrHelper(TF_Graph* graph, TF_Status* s,

func TestMigrateAutoIncrement(t *testing.T) {
	type AutoIncrementStruct struct {
		ID     int64   `gorm:"primarykey;autoIncrement"`
		Field1 uint32  `gorm:"column:field1"`
		Field2 float32 `gorm:"column:field2"`
	}

	if err := DB.AutoMigrate(&AutoIncrementStruct{}); err != nil {
		t.Fatalf("AutoMigrate err: %v", err)
	}

	const ROWS = 10
	for idx := 0; idx < ROWS; idx++ {