Merge changes from topic 'partial_tlab'

* changes:
  Fix some gtests
  Use partial TLAB regions