Back To Schedule
Thursday, October 18 • 10:00am - 10:30am
Graph Program Extraction and Device Partitioning in Swift for TensorFlow

Sign up or log in to save this to your schedule, view media, leave feedback and see who's attending!

Swift for Tensorflow (https://github.com/tensorflow/swift) is an Open Source project that provides a new way to develop machine learning models. It combines the usability/debuggability of imperative “define by run” programming models (like TensorFlow Eager and PyTorch) with the performance of TensorFlow session/XLA (graph compilation).

In this talk, we describe the design and implementation of deabstraction, Graph Program Extraction (GPE) and device partitioning used by Swift for TensorFlow. These algorithms rely on aggressive mid-level transformations that incorporate techniques including inlining, program slicing, interpretation, and advanced control flow analysis. While the initial application of these algorithms is to TensorFlow and machine learning, these algorithms may be applied to any domain that would benefit from an imperative definition of a computation graph, e.g. for high performance accelerators in other domains.

Thursday October 18, 2018 10:00am - 10:30am PDT
1 - General Session (Rm LL20ABC)