Simple logic changes in tpu
Hi @joesho112358 Can you please resolve conflicts? Thank you! This PR is stale because it has been open for 14 days with no activity. It will be closed if no further activity occurs. Thank you. This PR was closed because it has been inactive for 14 days since being marked as stale. Please reopen if you'd like to work on this further.
Noticed a couple things when reading through that I think will help with readability at least.