Diving into the TPU v2 and v3