Searched refs:index_flat (Results 1 – 1 of 1) sorted by relevance
712 auto index_flat = index->shaped<int64, 2>({batch.num_tasks(), 3}); in EmitIndexTensor() local716 index_flat(task_idx, 0) = task.guid; in EmitIndexTensor()717 index_flat(task_idx, 1) = offset; in EmitIndexTensor()718 index_flat(task_idx, 2) = offset + task.size(); in EmitIndexTensor()