Searched defs:per_partition_size (Results 1 – 4 of 4) sorted by relevance
88 const int64_t per_partition_size = in TopKWithPartitions() local
93 const int64_t per_partition_size = CeilOfRatio(input_size, shard_count); in HandleCustomCallTopK() local
1782 const int64_t per_partition_size = CeilOfRatio(input_size, partition_count); in HandleSort() local2354 const int64_t per_partition_size = partitioned_shape.dimensions(dim); in HandleDynamicUpdateSlice() local
1297 const int64_t per_partition_size = CeilOfRatio(input_size, shard_count); in GetKValueInTopKWhenPartitionSortDim() local