Searched refs:num_out_rows (Results 1 – 1 of 1) sorted by relevance
1279 int num_out_rows = (slice_num_cols + N - 1) / N * slice_num_rows; in ShuffleMatrix() local1280 DCHECK_LE(num_out_rows, buffer->dimension(0)); in ShuffleMatrix()1282 end = start + num_out_rows / i; in ShuffleMatrix()1284 num_out_rows -= (end - start); in ShuffleMatrix()