Home
last modified time | relevance | path

Searched defs:per_partition_size (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsorting.cc84 const int64 per_partition_size = CeilOfRatio(last_dim_size, num_partitions); in TopKWithPartitions() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc1665 const int64 per_partition_size = CeilOfRatio(input_size, partition_count); in HandleSort() local
1825 const int64 per_partition_size = CeilOfRatio(input_size, shard_count); in HandleCustomCall() local
2384 const int64 per_partition_size = partitioned_shape.dimensions(dim); in HandleDynamicUpdateSlice() local
Dspmd_partitioner_util.cc1251 const int64 per_partition_size = CeilOfRatio(input_size, shard_count); in GetKValueInTopKWhenPartitionSortDim() local