该论文介绍了基于TPUs上运行的完整张量程序的性能预测数据集TpuGraphs，该数据集提供了大规模图的图级预测任务，涉及规模可比较的图属性预测数据集的25倍的图，以及平均比现有的机器学习程序性能预测数据集大770倍的大型图。这带来了从可伸缩性、训练效率到模型质量等方面的新挑战。