Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsorting.cc88 const int64_t per_partition_size = in TopKWithPartitions() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dcustom_call_handler.cc93 const int64_t per_partition_size = CeilOfRatio(input_size, shard_count); in HandleCustomCallTopK() local
Dspmd_partitioner.cc1782 const int64_t per_partition_size = CeilOfRatio(input_size, partition_count); in HandleSort() local
2354 const int64_t per_partition_size = partitioned_shape.dimensions(dim); in HandleDynamicUpdateSlice() local
Dspmd_partitioner_util.cc1297 const int64_t per_partition_size = CeilOfRatio(input_size, shard_count); in GetKValueInTopKWhenPartitionSortDim() local