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());