Searched refs:slice_col_start (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 834 int slice_col_start, int slice_num_cols, const int N,
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()
1248 const int col_start = s / slice_num_rows * N + slice_col_start; in ShuffleMatrix()
1252 slice_col_start + slice_num_cols - 1); in ShuffleMatrix()
|