PatternRewriter &rewriter) const override {
    SmallVector<Type> result_types;
    result_types.reserve(op.getOutputs().size() + 1);
    for (const auto &output : op.getOutputs()) {
      Type ty = output.getType();
      if (auto tensor_ty = mlir::dyn_cast<RankedTensorType>(ty)) {
        if (!tensor_ty.hasStaticShape()) return failure();
      }