公开(公告)号：US20170124451A1

公开(公告)日：2017-05-04

申请号：US15336673

申请日：2016-10-27

Applicant: Google Inc.

Inventor： Paul Ronald Barham ,  Vijay Vasudevan

IPC: G06N3/08 ,  G06N99/00

CPC classification number: G06N3/08 ,  G06F9/5038 ,  G06F9/5066 ,  G06N3/0454 ,  G06N3/063 ,  G06N3/084 ,  G06N20/00

Abstract: Methods, systems, and apparatus, including computer programs encoded on computer storage media, for receiving, by a computational graph system, a request to process a computational graph; obtaining data representing a subgraph of the computational graph, the computational graph comprising a plurality of nodes and directed edges, wherein each node represents a respective operation, wherein each directed edge connects a respective first node to a respective second node, the subgraph assigned to a first device by a placer in the computational graph system; determining that the first device comprises a hardware accelerator having a plurality of streams; in response to determining, generating instructions that when executed by the first device cause the first device to: assign the operation represented by each node in the subgraph to a respective stream; and perform the operations represented by the nodes in the subgraph in accordance with the assignment.