This change requires the addition of an `fgraph` argument to the
`Op.infer_shape` method used by `ShapeFeature`.