ICLRApr, 2019

使用图神经网络模拟张量程序的执行时间

TL;DR本研究提出了一种基于抽象语法树的图形卷积网络的代理模型,以优化张量程序(如卷积)的执行时间。我们通过一个新的数据集来验证其有效性,并为运行时间预测任务提供了基线。