“A TPU's architecture is composed of a few large, coarse-grained units, such as large matrix units and a central vector unit.”