The CollagePartition
pass for finding optimal partitionings of Relay models.
See the RFC.
Based on:
Collage: Automated Integration of Deep Learning Backends Byungsoo Jeon, Sunghyun Park, Peiyuan Liao, Sheng Xu, Tianqi Chen, Zhihao Jia
CAUTION: This is a prototype, do not use in prod.