Automatically set input shapes by splitting the output shapes into smaller components based on a certain property.