Searched refs:indices_matrix (Results 1 – 3 of 3) sorted by relevance
237 auto indices_matrix = indices.matrix<int64>(); in GetNextInternal() local258 indices_matrix(current_position_in_values, 0) = i; in GetNextInternal()261 indices_matrix(current_position_in_values, k + 1) = in GetNextInternal()
152 auto indices_matrix = indices_out_->matrix<int64>(); in Update() local153 indices_matrix(output_index, 0) = batch_index; in Update()154 indices_matrix(output_index, 1) = cross_count; in Update()