Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsparse_fill_empty_rows_op.cc97 std::vector<Tindex> csr_offset(dense_rows, 0); in operator ()() local
104 ++csr_offset[row]; in operator ()()
111 bool row_empty = (csr_offset[row] == 0); in operator ()()
117 csr_offset[row] = std::max(csr_offset[row], Tindex{1}); in operator ()()
125 csr_offset[row] += csr_offset[row - 1]; in operator ()()
139 const Tindex N_full = csr_offset[dense_rows - 1]; in operator ()()
156 const Tindex output_i = ((row == 0) ? 0 : csr_offset[row - 1]) + offset; in operator ()()
170 const Tindex starting_index = (row == 0) ? 0 : csr_offset[row - 1]; in operator ()()