A Hierarchical Model for Device Placement

    Anna Goldie
    Anna Goldie
    Hieu Pham
    Hieu Pham
    Benoit Steiner
    Benoit Steiner

    international conference on learning representations, 2018.

    Cited by: 33|Bibtex|Views60|Links
    EI

    Abstract:

    We introduce a hierarchical model for efficient placement of computational graphs onto hardware devices, especially in heterogeneous environments with a mixture of CPUs, GPUs, and other computational devices. The algorithm learns to assign graph operations to groups and to allocate those groups to available devices. The grouping and devic...More

    Code:

    Data:

    Your rating :
    0

     

    Tags
    Comments