9.5.4 TPU v2/v3