Understanding the energy balance of TR structures observed by IRIS in non-equilibrium emission