added support for tokens_per_row for svd