TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs

25 August 2023

Papers citing "TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs"

8 / 8 papers shown

Title
A Benchmark on Directed Graph Representation Learning in Hardware Designs Haoyu Wang Yinan Huang Nan Wu Pan Li OOD 38 1 0 09 Oct 2024
GRANITE: A Graph Neural Network Model for Basic Block Throughput Estimation O. Sýkora P. Phothilimthana Charith Mendis Amir Yazdanbakhsh GNN 35 21 0 08 Oct 2022
Primer: Searching for Efficient Transformers for Language Modeling David R. So Wojciech Mañke Hanxiao Liu Zihang Dai Noam M. Shazeer Quoc V. Le VLM 83 149 0 17 Sep 2021
ProGraML: Graph-based Deep Learning for Program Optimization and Analysis Chris Cummins Zacharias V. Fisches Tal Ben-Nun Torsten Hoefler Hugh Leather 62 56 0 23 Mar 2020
Talking-Heads Attention Noam M. Shazeer Zhenzhong Lan Youlong Cheng Nan Ding L. Hou 89 79 0 05 Mar 2020
Chameleon: Adaptive Code Optimization for Expedited Deep Neural Network Compilation Byung Hoon Ahn Prannoy Pilligundla Amir Yazdanbakhsh H. Esmaeilzadeh ODL 53 80 0 23 Jan 2020
MoleculeNet: A Benchmark for Molecular Machine Learning Zhenqin Wu Bharath Ramsundar Evan N. Feinberg Joseph Gomes C. Geniesse Aneesh S. Pappu K. Leswing Vijay S. Pande OOD 157 1,748 0 02 Mar 2017
Neural Architecture Search with Reinforcement Learning Barret Zoph Quoc V. Le 264 5,290 0 05 Nov 2016