Fixed a bug involving shape_i op and constant folding. In short, new constants introduced by the constant folding optimization were not inserted in shape_of dictionary