for (auto& use : tensorlist.getUses()) {
      if (auto push = llvm::dyn_cast<TensorListPushBackOp>(use.getOwner())) {
        auto element_type =
            mlir::dyn_cast<RankedTensorType>(push.getTensor().getType());
        if (!verify_and_update_potential_element_type(element_type))
          return false;
        add_to_worklist(push.getOutputHandle());
        continue;
      }

  LogicalResult matchAndRewrite(OpTy op,
                                PatternRewriter &rewriter) const override {
    auto tensor_ty = mlir::dyn_cast<RankedTensorType>(op.getTensor().getType());
    auto indices_ty =
        mlir::dyn_cast<RankedTensorType>(op.getIndices().getType());
    auto updates_ty =
        mlir::dyn_cast<RankedTensorType>(op.getUpdates().getType());