Searched defs:size_per_partition (Results 1 – 1 of 1) sorted by relevance
56 int64_t size_per_partition = hlo->shape().dimensions().back(); in PadEachPartitionWithHaloExchange() local115 int64_t size_per_partition = hlo->shape().dimensions().back(); in ShuffleWithinEachPartitionUsingOneHot() local