The effort is part of Google's aggressive plan to make its Tensor Processing Units a viable alternative to Nvidia's market-leading GPUs