Searched refs:slice_row_start (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 833 const ConstMatrixMapR& mat, int slice_row_start, int slice_num_rows,
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()
1251 const auto* input_end = &mat(slice_row_start + slice_num_rows - 1, in ShuffleMatrix()
|