Searched refs:original_sharded_dot_shape (Results 1 – 1 of 1) sorted by relevance
937 auto original_sharded_dot_shape = original_dot->shape(); in PartitionBaseCase() local938 auto new_dot_shape = original_sharded_dot_shape; in PartitionBaseCase()997 HloInstruction::CreateReshape(original_sharded_dot_shape, dot)); in PartitionBaseCase()