TpuGraphs: 大型张量计算图的性能预测数据集

💡 原文中文,约300字,阅读约需1分钟。
📝

内容提要

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

🎯

关键要点

  • 该论文介绍了基于TPUs的性能预测数据集TpuGraphs。
  • TpuGraphs提供了大规模图的图级预测任务。
  • 该数据集的图规模是可比较的图属性预测数据集的25倍。
  • TpuGraphs的图平均比现有机器学习程序性能预测数据集大770倍。
  • 该数据集带来了可伸缩性、训练效率和模型质量等方面的新挑战。
➡️

继续阅读