add infer_shape test to as_op decorator