Searched refs:input_partitioned (Results 1 – 1 of 1) sorted by relevance
1778 auto input_partitioned = GetPartitionedHlo(hlo->operand(0)); in HandleCustomCall() local1779 if (!EvenlyPartitions(hlo->shape(), input_partitioned.sharding())) { in HandleCustomCall()1780 input_partitioned = input_partitioned.PadWithValue( in HandleCustomCall()1783 auto input = input_partitioned.hlo(); in HandleCustomCall()