Searched refs:mat_num_cols (Results 1 – 1 of 1) sorted by relevance
1144 const int mat_num_cols = transpose ? mat.dimension(0) : mat.dimension(1); in CreateSparseSlices() local1148 std::max(1, (mat_num_cols + slice_num_cols - 1) / slice_num_cols); in CreateSparseSlices()1169 std::min<int>(slice_num_cols, mat_num_cols - j * slice_num_cols); in CreateSparseSlices()1175 DSizes d(num_rows, mat_num_cols); in CreateSparseSlices()1252 const int mat_num_cols = mat.dimension(1); in ShuffleMatrix() local1253 const int row_slice_size = slice_num_rows * mat_num_cols; in ShuffleMatrix()1260 input_start += mat_num_cols; in ShuffleMatrix()1271 input_start += mat_num_cols; in ShuffleMatrix()