Home
last modified time | relevance | path

Searched refs:mat_num_cols (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc1143 const int mat_num_cols = transpose ? mat.dimension(0) : mat.dimension(1); in CreateSparseSlices() local
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()
1174 DSizes d(num_rows, mat_num_cols); in CreateSparseSlices()
1253 const int mat_num_cols = mat.dimension(1); in ShuffleMatrix() local
1254 const int row_slice_size = slice_num_rows * mat_num_cols; in ShuffleMatrix()
1261 input_start += mat_num_cols; in ShuffleMatrix()
1272 input_start += mat_num_cols; in ShuffleMatrix()