Energy partitioning and its controls over a heterogeneous semi-arid shrubland ecosystem in the Lake Naivasha Basin, Kenya