Searched refs:slice_num_rows (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 816 const ConstMatrixMapL& mat, bool transpose, int slice_num_rows,
834 const ConstMatrixMapR& mat, int slice_row_start, int slice_num_rows,
1140 int slice_num_rows, int slice_block_size, int slice_num_cols, in CreateSparseSlices() argument
1146 std::max(1, (mat_num_rows + slice_num_rows - 1) / slice_num_rows); in CreateSparseSlices()
1166 std::min<int>(slice_num_rows, mat_num_rows - i * slice_num_rows); in CreateSparseSlices()
1173 &mat(0, i * slice_num_rows), mat.dimensions()); in CreateSparseSlices()
1177 &mat(i * slice_num_rows, 0), d); in CreateSparseSlices()
1236 int slice_row_start, int slice_num_rows, int slice_col_start, in ShuffleMatrix() argument
1244 auto shuffle_work = [&mat, slice_row_start, slice_num_rows, slice_col_start, in ShuffleMatrix()
1246 const int row_start = s % slice_num_rows + slice_row_start; in ShuffleMatrix()
[all …]
|