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