Title
Polyhedral Optimization Of Tensorflow Computation Graphs
Abstract
We present R-Stream.TF, a polyhedral optimization tool for neural network computations. R-Stream.TF transforms computations performed in a neural network graph into C programs suited to the polyhedral representation and uses R-Stream, a polyhedral compiler, to parallelize and optimize the computations performed in the graph. R-Stream.TF can exploit the optimizations available with R-Stream to generate a highly optimized version of the computation graph, specifically mapped to the targeted architecture. During our experiments, R-Stream.TF was able to automatically reach performance levels close to the hand-optimized implementations, demonstrating its utility in porting neural network computations to parallel architectures.
Year
DOI
Venue
2017
10.1007/978-3-030-17872-7_5
PROGRAMMING AND PERFORMANCE VISUALIZATION TOOLS
DocType
Volume
ISSN
Conference
11027
0302-9743
Citations 
PageRank 
References 
1
0.36
0
Authors
5
Name
Order
Citations
PageRank
Benoît Pradelle1182.49
Benoît Meister213812.84
Muthu Manikandan Baskaran349333.10
Jonathan Springer4151.69
Richard Lethin511817.17