Searched refs:tmp_shard_shape (Results 1 – 1 of 1) sorted by relevance
2092 auto tmp_shard_shape = output_shard_shape; in HandleReshape() local2093 tmp_shard_shape.set_dimensions( in HandleReshape()2097 HloInstruction::CreateReshape(tmp_shard_shape, operand.hlo())); in HandleReshape()2100 auto tmp_full_shape = tmp_shard_shape; in HandleReshape()2102 output_sharded_dim, tmp_shard_shape.dimensions(output_sharded_dim) * in HandleReshape()2109 for (int64 i = 0; i < tmp_shard_shape.rank(); ++i) { in HandleReshape()2119 tmp_shard_shape.dimensions(output_sharded_dim) * in HandleReshape()