Searched refs:idx_parallel_tiles_num (Results 1 – 1 of 1) sorted by relevance
1842 int idx_parallel_tiles_num = new_index_shard.NumTiles(indices_parallel_dims); in GatherOperandsShardedAcrossParallelDims() local1844 if (idx_parallel_tiles_num == 1 && op_parallel_tiles_num == 1) { in GatherOperandsShardedAcrossParallelDims()1870 if (idx_parallel_tiles_num != op_parallel_tiles_num) { in GatherOperandsShardedAcrossParallelDims()1875 if (idx_parallel_tiles_num < op_parallel_tiles_num) { in GatherOperandsShardedAcrossParallelDims()