Searched refs:slice_num_rows (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 815 const ConstMatrixMapL& mat, bool transpose, int slice_num_rows,
833 const ConstMatrixMapR& mat, int slice_row_start, int slice_num_rows,
1139 int slice_num_rows, int slice_block_size, int slice_num_cols, in CreateSparseSlices() argument
1145 std::max(1, (mat_num_rows + slice_num_rows - 1) / slice_num_rows); in CreateSparseSlices()
1165 std::min<int>(slice_num_rows, mat_num_rows - i * slice_num_rows); in CreateSparseSlices()
1172 &mat(0, i * slice_num_rows), mat.dimensions()); in CreateSparseSlices()
1176 &mat(i * slice_num_rows, 0), d); in CreateSparseSlices()
1235 int slice_row_start, int slice_num_rows, int slice_col_start, in ShuffleMatrix() argument
1245 auto shuffle_work = [&mat, slice_row_start, slice_num_rows, slice_col_start, in ShuffleMatrix()
1247 const int row_start = s % slice_num_rows + slice_row_start; in ShuffleMatrix()
[all …]
|