Fixed Not implemented infer_shape.