Searched refs:slice_num_cols (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 816 int slice_block_size, int slice_num_cols,
834 int slice_col_start, int slice_num_cols, const int N,
1139 int slice_num_rows, int slice_block_size, int slice_num_cols, in CreateSparseSlices() argument
1147 std::max(1, (mat_num_cols + slice_num_cols - 1) / slice_num_cols); in CreateSparseSlices()
1168 std::min<int>(slice_num_cols, mat_num_cols - j * slice_num_cols); in CreateSparseSlices()
1182 [=]() { work(sparse_slice, slice, slice_num_cols * j); }); in CreateSparseSlices()
1236 int slice_num_cols, const int N, in ShuffleMatrix() argument
1246 slice_num_cols, N, buffer, counter](int s, int e) { in ShuffleMatrix()
1252 slice_col_start + slice_num_cols - 1); in ShuffleMatrix()
1256 const int aligned_end = slice_num_cols / N * slice_num_rows; in ShuffleMatrix()
[all …]
|