let results = (outs
    TF_Float32Tensor:$updated_embedding_table,
    TF_Float32Tensor:$updated_velocity,
    TF_Float32Tensor:$updated_momentum
  );
}

def TF_XlaSparseCoreFtrlOp : TF_Op<"XlaSparseCoreFtrl", []> {
  let summary = "aaa";

  let arguments = (ins
    TF_Float32Tensor:$embedding_table,
    TF_Float32Tensor:$accumulator,
    TF_Float32Tensor:$linear,
    TF_Float32Tensor:$learning_rate,