X-RLflow: Graph Reinforcement Learning for Neural Network Subgraphs Transformation

Guoliang He, Sean Parker, Eiko Yoneki

Tensor graph superoptimisation systems perform a sequence of subgraph substitution to neural networks, to find the optimal computation graph structure. Such a graph transformation…