Searched refs:spmd_operand (Results 1 – 1 of 1) sorted by relevance
1537 auto spmd_operand = GetPartitionedHlo(operand).Reshard(sharding).hlo(); in HandleConcatenate() local1543 spmd_operand->shape().dimensions(dimension), offset, 1) in HandleConcatenate()1548 temp_output_shape, temp_output, spmd_operand, start_indices)); in HandleConcatenate()