llvm::all_of(op->getResultTypes(), is_not_variant);
  };

  auto is_set_item_legal = [](Operation *op) {
    return op->hasAttr("resize_if_index_out_of_bounds") &&
           op->getAttr("resize_if_index_out_of_bounds")
               .cast<mlir::BoolAttr>()
               .getValue();
  };

  ConversionTarget target(*context);