公开(公告)号：US20200293838A1

公开(公告)日：2020-09-17

申请号：US16818932

申请日：2020-03-13

Applicant: DeepMind Technologies Limited

Inventor： Yujia Li ,  Vinod Nair ,  Felix Axel Gimeno Gil ,  Aditya Paliwal ,  Miles C. Lubin

IPC: G06K9/62 ,  G06N3/08 ,  G06N5/04 ,  G06N20/10

Abstract: Methods, systems, and apparatus, including computer programs encoded on computer storage media, for generating a schedule for a computation graph. One of the methods includes obtaining data representing an input computation graph; processing the data representing the input computation graph using a graph neural network to generate one or more instance-specific proposal distributions; and generating a schedule for the input computation graph by performing an optimization algorithm in accordance with the one or more instance-specific proposal distributions.