Searched refs:dense_rows (Results 1 – 2 of 2) sorted by relevance
56 const Tindex dense_rows = dense_shape(0); in operator ()() local62 TensorShape({dense_rows}), in operator ()()76 if (dense_rows == 0) { in operator ()()97 std::vector<Tindex> csr_offset(dense_rows, 0); in operator ()()100 if (row < 0 || row >= dense_rows) { in operator ()()102 " >= ", dense_rows); in operator ()()109 for (int row = 0; row < dense_rows; ++row) { in operator ()()139 const Tindex N_full = csr_offset[dense_rows - 1]; in operator ()()150 std::vector<Tindex> filled_count(dense_rows, 0); in operator ()()167 for (Tindex row = 0; row < dense_rows; ++row) { in operator ()()
842 dense_rows = []858 dense_rows.append(array_ops.concat(row_blocks, axis=-1))860 mat = array_ops.concat(dense_rows, axis=-2)