Given an env and the shape of all inputs returns a dictionary holding the
shapes of all intermediate and output variables