TpuGraphs: 大型张量计算图的性能预测数据集
💡
原文中文,约300字,阅读约需1分钟。
📝
内容提要
该论文介绍了基于TPUs上运行的完整张量程序的性能预测数据集TpuGraphs,该数据集提供了大规模图的图级预测任务,涉及规模可比较的图属性预测数据集的25倍的图,以及平均比现有的机器学习程序性能预测数据集大770倍的大型图。这带来了从可伸缩性、训练效率到模型质量等方面的新挑战。
🎯
关键要点
- 该论文介绍了基于TPUs的性能预测数据集TpuGraphs。
- TpuGraphs提供了大规模图的图级预测任务。
- 该数据集的图规模是可比较的图属性预测数据集的25倍。
- TpuGraphs的图平均比现有机器学习程序性能预测数据集大770倍。
- 该数据集带来了可伸缩性、训练效率和模型质量等方面的新挑战。
➡️