Searched refs:indices_matrix (Results 1 – 2 of 2) sorted by relevance
245 auto indices_matrix = indices.matrix<int64_t>(); in GetNextInternal() local266 indices_matrix(current_position_in_values, 0) = i; in GetNextInternal()269 indices_matrix(current_position_in_values, k + 1) = in GetNextInternal()
288 auto indices_matrix = indices_out_->matrix<int64_t>(); in Update() local289 indices_matrix(output_index, 0) = batch_index; in Update()290 indices_matrix(output_index, 1) = cross_count; in Update()