Support row wise wieghts mode in csr_row_mode