Searched refs:out_indices_mat (Results 1 – 3 of 3) sorted by relevance
285 typename TTypes<int64>::Matrix out_indices_mat = in Compute() local288 out_indices_mat.setZero(); in Compute()305 out_indices_mat(i, reduction.group_by_dims[j]) = group[j]; in Compute()307 out_indices_mat(i, j) = group[j]; in Compute()
153 auto out_indices_mat = out_indices_t->matrix<int64>(); in Compute() local159 out_indices_mat.chip<0>(i) = in Compute()
154 auto out_indices_mat = out_indices_t->matrix<int64>(); in OutputSparseTensor() local174 out_indices_mat(value_index, i) = group_indices[i]; in OutputSparseTensor()176 out_indices_mat(value_index, group_indices.size()) = group_value_index; in OutputSparseTensor()