I think tech companies are doing layoffs partly because they need to cover AI operating expenses.

I think so too, otherwise why wouldn't you put that (purported) increased capacity/output into improving your existing products or creating new ones, with the headcount that you already have?