EFFICIENT HARDWARE ACCELERATOR ARCHITECTURE EXPLORATION

摘要

Methods, systems, and apparatus, including computer programs encoded on computer storage media, for determining architectures of hardware accelerators. In one aspect, a method includes receiving data specifying a plurality of hardware parameters; receiving data specifying one or more predetermined values for each of one or more of the plurality of hardware parameters; generating a plurality of candidate hardware architectures that are specific to a particular machine learning task by repeatedly performing the following operations: selecting a respective value for each of the plurality of hardware parameters; determining a candidate hardware architecture; determining whether the candidate hardware architecture satisfies pre-evaluation criteria; and in response to a positive determination, evaluating a performance measure of the candidate hardware architecture on the particular machine learning task; and generating a final hardware architecture based on the plurality of candidate hardware architectures and on the performance measures.