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