Searched refs:temp_output_shape (Results 1 – 1 of 1) sorted by relevance
1512 auto temp_output_shape = MakePartitionedShape(hlo->shape(), sharding); in HandleConcatenate() local1521 int temp_output_padding = temp_output_shape.dimensions(dimension) * in HandleConcatenate()1528 temp_output_shape.set_dimensions( in HandleConcatenate()1529 dimension, temp_output_shape.dimensions(dimension) * in HandleConcatenate()1532 auto temp_output = CreateZero(temp_output_shape, &b_); in HandleConcatenate()1548 temp_output_shape, temp_output, spmd_operand, start_indices)); in HandleConcatenate()