tpu_utils
